1 files changed, 34 insertions, 0 deletions
diff --git a/test/test_files.py b/test/test_files.py
new file mode 100644
index 00000000..4852d692
--- /dev/null
+++ b/test/test_files.py
@@ -0,0 +1,34 @@
+"""Tests for files.py"""
+
+import os, sys
+
+from coverage.files import FileLocator
+
+sys.path.insert(0, os.path.split(__file__)[0]) # Force relative import for Py3k
+from coveragetest import CoverageTest
+
+
+class FileLocatorTest(CoverageTest):
+    """Tests of `FileLocator`."""
+    
+    def abs_path(self, p):
+        return os.path.join(os.getcwd(), os.path.normpath(p))
+
+    def test_simple(self):
+        self.make_file("hello.py", "#hello")
+        fl = FileLocator()
+        self.assertEqual(fl.relative_filename("hello.py"), "hello.py")
+        abs = self.abs_path("hello.py")
+        self.assertNotEqual(abs, "hello.py")
+        self.assertEqual(fl.relative_filename(abs), "hello.py")
+
+    def test_peer_directories(self):
+        self.make_file("sub/proj1/file1.py", "file1")
+        self.make_file("sub/proj2/file2.py", "file2")
+        abs1 = self.abs_path("sub/proj1/file1.py")
+        abs2 = self.abs_path("sub/proj2/file2.py")
+        d = os.path.normpath("sub/proj1")
+        os.chdir(d)
+        fl = FileLocator()
+        self.assertEqual(fl.relative_filename(abs1), "file1.py")
+        self.assertEqual(fl.relative_filename(abs2), abs2)