diff options
-rwxr-xr-x | test/test.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/test/test.py b/test/test.py index c8c0e80..6f0f496 100755 --- a/test/test.py +++ b/test/test.py @@ -58,10 +58,14 @@ class MagicTest(unittest.TestCase): def test_from_buffer_str_and_bytes(self): m = magic.Magic(mime=True) - s = '#!/usr/bin/env python\nprint("foo")' - self.assertEqual("text/x-script.python", m.from_buffer(s)) - b = b'#!/usr/bin/env python\nprint("foo")' - self.assertEqual("text/x-script.python", m.from_buffer(b)) + + self.assertTrue( + m.from_buffer('#!/usr/bin/env python\nprint("foo")') + in ("text/x-python", "text/x-script.python")) + self.assertTrue( + m.from_buffer(b'#!/usr/bin/env python\nprint("foo")') + in ("text/x-python", "text/x-script.python")) + def test_mime_types(self): dest = os.path.join(MagicTest.TESTDATA_DIR, |