Comping

tests/test_parts.py

@@ -40,6 +40,8 @@ class PartGenerationTests(unittest.TestCase):
       generator.model.validate_document(result)
       for measure in result['staves'][0]['measures']:
         events = measure['voices'][0]['notes']
+        if measure.get('ending_sustain'):
+          continue
         if measure.get('section_label') in ('Rhythm Lead-in', 'Guitar Melody', 'Acoustic Solo', 'Electric Solo'):
           continue
         self.assert_complete_measure(measure)
@@ -91,7 +93,7 @@ class PartGenerationTests(unittest.TestCase):
     source = tests.test_musicxml.parse(xml.etree.ElementTree.tostring(root, encoding='unicode'))
     before = copy.deepcopy(source)
     expected = [n for m in source['staves'][0]['measures'] for n in m['voices'][0]['notes']]
-    for builder in (tests.arrangement_fixtures.section_document, generator.arrange.generate.arrange_document):
+    for builder in (tests.arrangement_fixtures.section_document, tests.arrangement_fixtures.melody_performance):
       for part in ('electric', 'vocal'):
         with self.subTest(builder=builder.__module__, part=part):
           result = builder(source, part)