fixed issue with single length relative paths

DomWeldon · Sep 21, 2017 · 0cb7df9 · 0cb7df9
1 parent 875315e
commit 0cb7df9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/oopycql/query.py b/oopycql/query.py
@@ -68,6 +68,8 @@ def __new__(cls, *args, **kwargs):
                 except TypeError:
                     prepend_path = Path('../')
             filename = reduce(lambda x, y:  x / Path(y), fqn.split('.'))
+            if type(filename) == str:
+                filename = Path(filename)
             if prepend_path is not None:
                 filename = prepend_path / filename
             if version_info.major == 2 \

diff --git a/tests/test_relative_import/test_query.cql b/tests/test_relative_import/test_query.cql
@@ -0,0 +1 @@
+MATCH (n) RETURN COUNT(n)
diff --git a/tests/test_relative_import/test_relative_import.py b/tests/test_relative_import/test_relative_import.py
@@ -11,3 +11,7 @@ class RelativeImportTestCase(TestCase):
     def test_relative_import(self):
         cq = CypherQuery('..cql.from_module')
         assert str(cq) == 'MATCH (n) RETURN COUNT(n)\n'
+
+    def test_relative_import_same_directory(self):
+        cq = CypherQuery('.test_query')
+        assert str(cq) == 'MATCH (n) RETURN COUNT(n)\n'