diff --git a/src/fake_bpy_module/generator.py b/src/fake_bpy_module/generator.py index 4bc76b74b..2dde436c5 100644 --- a/src/fake_bpy_module/generator.py +++ b/src/fake_bpy_module/generator.py @@ -673,7 +673,8 @@ def build_child_modules( if module_level != 0: filename = re.sub(r"\.", "/", mod_name) + ".py" else: - filename = re.sub(r"\.", "/", mod_name) + "/__init__.py" + filename = \ + re.sub(r"\.", "/", mod_name) + "/__init__.py" info = gen_info.create_target(filename) info.data = [] info.child_modules = [] diff --git a/tests/python/fake_bpy_module_test/fake_bpy_module_test/generator_test.py b/tests/python/fake_bpy_module_test/fake_bpy_module_test/generator_test.py index 88ad89877..266e04a1e 100644 --- a/tests/python/fake_bpy_module_test/fake_bpy_module_test/generator_test.py +++ b/tests/python/fake_bpy_module_test/fake_bpy_module_test/generator_test.py @@ -634,9 +634,11 @@ def test_single_rule(self): self.assertIsNotNone(actual_rule) self.assertIsNotNone(actual_gen_info) - self.assertEqual(set(actual_gen_info.targets()), {"module_abc/__init__.py"}) + self.assertEqual( + set(actual_gen_info.targets()), {"module_abc/__init__.py"}) - target_module_abc = actual_gen_info.get_target("module_abc/__init__.py") + target_module_abc = actual_gen_info.get_target( + "module_abc/__init__.py") self.assertEqual(len(target_module_abc.data), 1) self.assertEqual(target_module_abc.data[0].type(), "class") self.assertEqual(target_module_abc.data[0].name(), "Class123") @@ -726,7 +728,8 @@ def test_multiple_rules(self): self.assertEqual( set(actual_gen_info_1.targets()), {"module_1/__init__.py", "module_1/submodule_1.py"}) - self.assertEqual(set(actual_gen_info_2.targets()), {"module_2/__init__.py"}) + self.assertEqual( + set(actual_gen_info_2.targets()), {"module_2/__init__.py"}) target_module_1 = actual_gen_info_1.get_target("module_1/__init__.py") self.assertEqual(len(target_module_1.data), 1) @@ -824,7 +827,8 @@ def test_exceptional_rule(self): self.assertEqual( set(actual_gen_info.targets()), {"module_exceptional/__init__.py"}) - target_module_abc = actual_gen_info.get_target("module_exceptional/__init__.py") + target_module_abc = actual_gen_info.get_target( + "module_exceptional/__init__.py") self.assertEqual(len(target_module_abc.data), 2) self.assertEqual(target_module_abc.data[0].type(), "class") self.assertEqual(target_module_abc.data[0].name(), "ClassExp")