add stdout assertion to test_app

DiamondLightSource · Dec 19, 2024 · 6352b90 · 6352b90
1 parent 94c3f97
commit 6352b90
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -19,9 +19,10 @@ def test_app():
         ) as mock_calculate_optimal_voltages,
     ):
         mock_calculate_optimal_voltages.return_value = np.array([72.14, 50.98, 18.59])
-        runner.invoke(app, ["calculate-voltages", "tests/data/raw_data.csv"])
+        result = runner.invoke(app, ["calculate-voltages", "tests/data/raw_data.csv"])
         mock_np_save.assert_called_once()
         mock_calculate_optimal_voltages.assert_called_with("tests/data/raw_data.csv")
+        assert "The optimal voltages are: [72.14, 50.98, 18.59]" in result.stdout
 
 
 def test_cli_version():