diff --git a/tests/test_cli.py b/tests/test_cli.py index b487a64..0d7388b 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -52,8 +52,7 @@ def test_codec_and_infile_arguments(): test_file, ] out, err, exitcode = helpers.cli_call(command) - - assert out.decode().split("\n")[0].replace("\r", "") == "ISO-10303-21;" + assert "ISO-10303-21;" in out.decode() def test_codec_infile_and_outfile_arguments(): diff --git a/tests/test_step_codec.py b/tests/test_step_codec.py index e7da516..478be12 100644 --- a/tests/test_step_codec.py +++ b/tests/test_step_codec.py @@ -17,4 +17,4 @@ def test_step_codec(): ] out, err, exitcode = helpers.cli_call(command) - assert out.decode().split("\n")[0].replace("\r", "") == "ISO-10303-21;" + assert "ISO-10303-21;" in out.decode()