Add ANSI stripping function and update assertions for JSON error messages

Author: msabramo

tests/test_cli.py

@@ -4,6 +4,7 @@
 from typer.testing import CliRunner
 from unittest import mock
 from firefly.cli import app, mock_image
+import re
 
 runner = CliRunner()
 
@@ -189,6 +190,10 @@ def test_generate_with_all_new_options(monkeypatch):
     assert result.exit_code != 0
     assert "content_class must be either 'photo' or 'art'" in result.output
 
+def strip_ansi(text):
+    ansi_escape = re.compile(r'\x1b\[[0-9;]*[mGKHF]')
+    return ansi_escape.sub('', text)
+
 def test_generate_invalid_json_style(monkeypatch):
     result = runner.invoke(
         app,
@@ -202,7 +207,8 @@ def test_generate_invalid_json_style(monkeypatch):
         ]
     )
     assert result.exit_code == 2
-    assert "Invalid JSON for --style" in result.output
+    assert "Invalid JSON for --style" in strip_ansi(result.output)
+
 
 def test_generate_invalid_json_structure(monkeypatch):
     result = runner.invoke(
@@ -217,7 +223,7 @@ def test_generate_invalid_json_structure(monkeypatch):
         ]
     )
     assert result.exit_code == 2
-    assert "Invalid JSON for --structure" in result.output
+    assert "Invalid JSON for --structure" in strip_ansi(result.output)
 
 def test_generate_invalid_num_variations(monkeypatch):
     # Test too low