support changed mime types in test

ahupp · ahupp · commit 8d8e4858af4e · 2020-09-15T00:23:52.000-07:00
diff --git a/test/test.py b/test/test.py
@@ -58,10 +58,14 @@ def test_from_file_str_and_bytes(self):
 
     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,
