Merge pull request #46 from athul/fix-tests

fix: fixes import path

.gitignore

@@ -1,2 +1,5 @@
 #generated when running the tests
-__pycache__/
+__pycache__/
+
+# Local VS code configurations
+.vscode/

tests/test_main.py

@@ -1,10 +1,15 @@
 '''
 Tests for the main.py
 '''
-from main import make_graph
 import unittest
 import os
 
+try:
+    # For travis build which uses
+    # python -m unittest discover
+    from main import make_graph
+except Exception as e:
+    print("Error: missing 'main.py'\nTrying ablsolute import...")
 
 class TestMain(unittest.TestCase):
 
@@ -47,4 +52,15 @@ class TestMain(unittest.TestCase):
 
 
 if __name__ == '__main__':
+    if __package__ is None:
+        import sys
+        # because test_main.py is one level below main.py
+        # python test/test_main.py
+        sys.path.append(os.path.dirname(
+        os.path.dirname(os.path.abspath(__file__))))
+        from main import make_graph
+    else:
+        # Later on if WakaReadme is implemetaion as package
+        # python -m tests/test_main
+        from ..main import make_graph
     unittest.main()