Test for get_encoding_file

sebastinas · sebastinas · commit 7587ad8dfeab · 2015-02-07T22:16:00.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/test/test_inspection.py b/bpython/test/test_inspection.py
@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 
+import os
+
 try:
     import unittest2 as unittest
 except ImportError:
@@ -102,5 +104,20 @@ def test_get_source_latin1(self):
         self.assertEqual(inspection.get_source_unicode(encoding_latin1.foo),
                          foo_non_ascii)
 
+    def test_get_source_file(self):
+        path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
+                            'fodder')
+
+        encoding = inspection.get_encoding_file(
+            os.path.join(path, 'encoding_ascii.py'))
+        self.assertEqual(encoding, 'ascii')
+        encoding = inspection.get_encoding_file(
+            os.path.join(path, 'encoding_latin1.py'))
+        self.assertEqual(encoding, 'latin1')
+        encoding = inspection.get_encoding_file(
+            os.path.join(path, 'encoding_utf8.py'))
+        self.assertEqual(encoding, 'utf-8')
+
+
 if __name__ == '__main__':
     unittest.main()
