add travis

Author: tangwei94

.travis.yml

@@ -0,0 +1,10 @@
+language: python
+python:
+  - "3.6"
+  - "3.7"
+# command to install dependencies
+install:
+  - pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+  - pip install -r requirements.txt
+script:
+  - pytest

requirements.txt

@@ -0,0 +1 @@
+numpy os io subprocess sys

test.sh

@@ -1 +1 @@
-env PYTHONPATH=$(pwd):${PYTHONPATH} pytest -vv -l --disable-pytest-warnings
+pytest -vv -l --disable-pytest-warnings

tests/test_cmpo.py

@@ -1,5 +1,8 @@
 import torch
 import numpy as np
+import sys, os
+testdir = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(testdir+"/..")
 from cmpo import *
 
 def test_project():

tests/test_model.py

@@ -1,8 +1,12 @@
 import torch
 import numpy as np
+import functools
+import sys, os
+testdir = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(testdir+"/..")
 from cmpo import *
 from model import *
-import functools
+
 tnp = lambda a, b, c: torch.einsum('ab,cd,ef->acebdf', a, b, c).view(a.shape[0]*b.shape[0]*c.shape[0], a.shape[1]*b.shape[1]*c.shape[1])
 
 def decorator_test_model(func):