Add Travis build with pystencils from pip

.travis.yml

@@ -14,11 +14,18 @@ addons:
     update: true
 matrix:
   fast_finish: true
+  allow_failures:
+    - name: "Pystencils from pip"
   include:
     - python: 3.6
       env: DISTRIB="ubuntu" TOX_PYTHON_VERSION="py36" COVERAGE="true" LINT="false"
       before_install:
         - sudo apt-get install -y ninja-build build-essential nvidia-cuda-toolkit
+    - python: 3.6
+      name: "Pystencils from pip"
+      env: DISTRIB="ubuntu" TOX_PYTHON_VERSION="py36" COVERAGE="true" LINT="false" PYSTENCIL_FROM_PIP="true"
+      before_install:
+        - sudo apt-get install -y ninja-build build-essential nvidia-cuda-toolkit
     - name: "Conda"
       env: DISTRIB="conda" PYTHON_VERSION="3.6" COVERAGE="false" LINT="false"
       before_install:

tests/backends/test_torch_native_compilation.py

@@ -194,12 +194,3 @@ def test_execute_torch_gpu():
     print(y_tensor)
     print(rtn)
 
-
-def main():
-    test_jit()
-    test_torch_native_compilation()
-    # test_generate_torch()
-    # test_execute_torch()
-
-
-main()

tests/test_autodiff.py

@@ -50,11 +50,3 @@ def test_simple_2d_check_raw_assignments():
         pystencils_autodiff.create_backward_assignments(
             forward_assignments, diff_mode=diff_mode)
 
-
-def main():
-    test_simple_2d_check_assignment_collection()
-    test_simple_2d_check_raw_assignments()
-
-
-if __name__ == '__main__':
-    main()

tests/test_tfmad.py

@@ -247,14 +247,3 @@ def test_tfmad_two_outputs():
     print(curl_op.backward_assignments)
 
 
-def main():
-    # test_tfmad_stencil()
-    # test_tfmad_two_stencils()
-    # test_tfmad_gradient_check_torch()
-    test_tfmad_gradient_check()
-    # test_tfmad_vector_input_data()
-    # test_tfmad_two_outputs()
-
-
-if __name__ == '__main__':
-    main()

tests/travis_install.sh

@@ -45,8 +45,12 @@ pip install -U pip setuptools
 pip install tox
 pip install codecov
 pip install sphinx
-# use newest pystencils
-pip install git+https://github.com/mabau/pystencils.git
+
+if [[ -z "$PYSTENCIL_FROM_PIP" ]]; then
+    pip install git+https://github.com/mabau/pystencils.git
+else
+    pip install pystencils
+fi
 pip install flake8
 
 if [[ "$COVERAGE" == "true" ]]; then