Cleanup

tests/test_solo_comparison.py

@@ -7,10 +7,13 @@ from pathlib import Path
 import tempfile
 from types import SimpleNamespace
 import unittest
+from unittest.mock import patch
+import contextlib
+import io
 
 from audition import reference_review, solo_compare
 from generator import vocabulary, production
-from generator.solo import export, harmonic, notation, placement, routes, types
+from generator.solo import export, harmonic, notation, placement, types
 from tests.solo_fixtures import song
 from tests.solo_fixtures import note
 from tests.solo_fixtures import choices
@@ -26,6 +29,17 @@ def pitched(duration, tie='', octave=5):
 
 
 class ComparisonTests(unittest.TestCase):
+  def test_exact_cli_is_a_real_exit_gate(self):
+    args=['solo_compare','--generated-dir','unused','--output','unused','--reference','fixture/acoustic=unused','--require-exact']
+    review={'exact_musical_match':False,'exact_matching_attacks':0,'differences':[{}],'rest_difference':True}
+    with patch('sys.argv',args),patch.object(solo_compare,'run',return_value={'solos':{'fixture/acoustic':review}}),contextlib.redirect_stdout(io.StringIO()):
+      self.assertEqual(solo_compare.main(),1)
+      review.update(exact_musical_match=True,differences=[],rest_difference=False)
+      self.assertEqual(solo_compare.main(),0)
+    with patch('sys.argv',args[:-3]+['--require-exact']),contextlib.redirect_stderr(io.StringIO()):
+      with self.assertRaises(SystemExit) as error:solo_compare.main()
+      self.assertEqual(error.exception.code,2)
+
   def test_note_container_does_not_turn_color_label_into_fingering(self):
     with tempfile.TemporaryDirectory() as root:
       note_xml=pitched(24).replace('</note>','<notations><technical><fingering>6</fingering></technical></notations></note>')