Apply change from Corin-EU to allow tests to be run from any directory

ahupp · ahupp · commit 41afcd5eab52 · 2018-08-16T08:09:17.000-07:00
diff --git a/test/test.py b/test/test.py
@@ -9,7 +9,7 @@
 
 
 class MagicTest(unittest.TestCase):
-    TESTDATA_DIR = os.path.join(os.path.dirname(__file__), 'testdata')
+    TESTDATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'testdata')
 
     def assert_values(self, m, expected_values, buf_equals_file=True):
         for filename, expected_value in expected_values.items():
@@ -34,10 +34,12 @@ def assert_values(self, m, expected_values, buf_equals_file=True):
                 self.assertIn(value, expected_value)
 
     def test_from_file_str_and_bytes(self):
+        filename = os.path.join(self.TESTDATA_DIR, "test.pdf")
+
         self.assertEqual('application/pdf',
-                         magic.from_file("test/testdata/test.pdf", mime=True))
+                         magic.from_file(filename, mime=True))
         self.assertEqual('application/pdf',
-                         magic.from_file(b"test/testdata/test.pdf", mime=True))
+                         magic.from_file(filename.encode('utf-8'), mime=True))
 
     def test_from_buffer_str_and_bytes(self):
         m = magic.Magic(mime=True)
