Cleanup

tests/test_catalog.py

@@ -1,10 +1,10 @@
+import generator.production
 import tests.production_fixtures
 import json
 import pathlib
 import tempfile
 import unittest
 
-import generator.arrange.generate
 import tests.solo_fixtures
 import tests.test_musicxml
 
@@ -16,7 +16,7 @@ class CatalogTests(unittest.TestCase):
       source, output = pathlib.Path(temporary) / 'songs', pathlib.Path(temporary) / 'output'
       source.mkdir()
       (source / 'song.xml').write_text(tests.test_musicxml.xml_score('<harmony><root><root-step>E</root-step><root-alter>-1</root-alter></root><kind>major</kind></harmony>' + tests.test_musicxml.xml_note('E', alter=-1), '<key><fifths>-3</fifths></key>'))
-      self.assertEqual(generator.arrange.generate.generate_performance_scores(source, output), 3)
+      self.assertEqual(generator.production.generate_tree(source, output), 3)
       self.assertEqual({p.name for p in output.iterdir()}, {'song_acoustic.json', 'song_electric.json', 'song_vocal.json'})
       for path in output.iterdir():
         metadata = json.loads(path.read_text())['metadata']