simplify argument usage in unit test

googleapis · Feb 28, 2024 · aba703a · aba703a
1 parent b8bdd31
commit aba703a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/library_generation/test/unit_tests.py b/library_generation/test/unit_tests.py
@@ -378,8 +378,7 @@ def test_get_library_returns_api_shortname(self):
         self.assertEqual("secretmanager", util.get_library_name(library_2))
 
     def test_generate_prerequisite_files_non_monorepo_success(self):
-        num_libraries = 1
-        library_path = self.__setup_prerequisite_files(num_libraries)
+        library_path = self.__setup_prerequisite_files(num_libraries=1)
 
         self.__compare_files(
             f"{library_path}/.repo-metadata.json",
@@ -396,8 +395,7 @@ def test_generate_prerequisite_files_non_monorepo_success(self):
         )
 
     def test_generate_prerequisite_files_monorepo_success(self):
-        num_libraries = 2
-        library_path = self.__setup_prerequisite_files(num_libraries)
+        library_path = self.__setup_prerequisite_files(num_libraries=2)
 
         self.__compare_files(
             f"{library_path}/.repo-metadata.json",