tests: fix test_entry_point() when there are several monitors

Author: BoboTiG

CHANGELOG

@@ -6,6 +6,7 @@ History:
       - Mac: reduce the number of function calls
       - Mac: support macOS Big Sur (fixes #178)
       - tests: fix macOS intepreter not found on Travis-CI
+      - tests: fix test_entry_point() when there are several monitors
       - :heart: contributors: @
 
 6.0.0   2020/06/30

mss/tests/test_implementation.py

@@ -96,11 +96,12 @@ def test_entry_point(capsys, sct):
     fmt = "sct-{width}x{height}.png"
     for opt in ("-o", "--out"):
         main([opt, fmt])
-        filename = fmt.format(**sct.monitors[1])
         out, _ = capsys.readouterr()
-        assert out.endswith(filename + "\n")
-        assert os.path.isfile(filename)
-        os.remove(filename)
+        for monitor, line in zip(sct.monitors[1:], out.splitlines()):
+            filename = fmt.format(**monitor)
+            assert line.endswith(filename)
+            assert os.path.isfile(filename)
+            os.remove(filename)
 
     fmt = "sct_{mon}-{date:%Y-%m-%d}.png"
     for opt in ("-o", "--out"):