Training Model

tests/test_catalog.py

@@ -5,8 +5,6 @@ import tempfile
 import unittest
 
 import generator.arrange.generate
-import tests.rhythm_fixtures
-import tests.scale_fixtures
 import tests.solo_fixtures
 import tests.test_musicxml
 
@@ -18,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(tests.test_musicxml.xml_note('E', alter=-1), '<key><fifths>-3</fifths></key>'))
-      self.assertEqual(generator.arrange.generate.generate_performance_scores(source, output, rhythms=tests.rhythm_fixtures.library(), scales=tests.scale_fixtures.library(), sequences=tests.solo_fixtures.library()), 3)
+      self.assertEqual(generator.arrange.generate.generate_performance_scores(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']