From 8d8e4858af4e9b157f8e83cd8f1076c36ca6338d Mon Sep 17 00:00:00 2001 From: Adam Hupp Date: Tue, 15 Sep 2020 00:22:42 -0700 Subject: support changed mime types in test --- test/test.py | 12 ++++++++---- 1 file 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, -- cgit v1.2.1