fix expected output string

pitt-crc · May 7, 2024 · b6cd7cc · b6cd7cc
1 parent 9d0103f
commit b6cd7cc
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_crc_sus.py b/tests/test_crc_sus.py
@@ -30,11 +30,9 @@ class OutputStringFormatting(TestCase):
     def test_output_matches_string(self) -> None:
         """Compare output string from the app with manually constructed expectation"""
 
-        output_string = CrcSus().build_output_string(account='sam', smp=10, htc=20)
+        output_string = CrcSus().build_output_string(account='sam', used=0, total=10, cluster='SMP')
         expected_string = (
-            'Account sam\n'
-            ' cluster smp has 10 SUs remaining\n'
-            ' cluster htc has 20 SUs remaining'
+            'Account sam\n cluster SMP has 10 SUs remaining'
         )
 
         self.assertEqual(expected_string, output_string)