WIP

nutti · Oct 1, 2023 · 6e4886c · 6e4886c
1 parent b130e5f
commit 6e4886c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git 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
@@ -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")