fix matplotlib travis error

.travis.yml

@@ -27,6 +27,6 @@ before_script:
 install: source travis/install.sh
 
 script:
- - python -m unittest discover
+ - make unittest
 
 after_script: conda remove --name xenonpy --all

Makefile

@@ -0,0 +1,10 @@
+export SHELL := /bin/bash
+
+doctest:
+	pytest --doctest-modules xenonpy
+
+unittest:
+	pytest tests
+
+lint:
+	pylint xenonpy

requirements.txt

@@ -1,12 +1,10 @@
-matplotlib~=2.1.1
-numpy~=1.14.0
-pandas~=0.22.0
+matplotlib
+numpy
+pandas
 pymatgen
 PyMonad
 PyYAML
-scikit-learn~=0.19.1
-scipy~=1.0
+scikit-learn
+scipy
 seaborn
-Sphinx
-sphinx-rtd-theme
-urllib3==1.22
+urllib3

tests/models/test_layer.py

@@ -2,3 +2,7 @@
 # Use of this source code is governed by a BSD-style
 # license that can be found in the LICENSE file.
 
+
+def test_layer():
+    print('test')
+    assert 1

tests/test_datasets.py

@@ -2,30 +2,30 @@
 # Use of this source code is governed by a BSD-style
 # license that can be found in the LICENSE file.
 
-import unittest
+import pytest as pt
 
+@pt.fixture(scope='module')
+def setup():
+    print('ignore NumPy RuntimeWarning\n')
+    import warnings
+    warnings.filterwarnings("ignore", message="numpy.dtype size changed")
+    warnings.filterwarnings("ignore", message="numpy.ndarray size changed")
+    yield 'running test'
+    print('test over')
 
-class TestDatasets(unittest.TestCase):
-    '''Test dataset module'''
 
-    @classmethod
-    def setUpClass(cls):
-        print('ignore NumPy RuntimeWarning\n')
-        import warnings
-        warnings.filterwarnings("ignore", message="numpy.dtype size changed")
-        warnings.filterwarnings("ignore", message="numpy.ndarray size changed")
+# def test_load(self):
+#     '''Test load function'''
+#     e = load('elements')
+#     self.assertEqual(118, e.shape[0])
+#     self.assertEqual(74, e.shape[1])
+#     e = load('mp_inorganic')
+#     self.assertEqual(69640, e.shape[0])
+#     self.assertEqual(15, e.shape[1])
+#     e = load('electron_density')
+#     self.assertEqual(103, e.shape[0])
+#     self.assertEqual(401, e.shape[1])
 
-    # def test_load(self):
-    #     '''Test load function'''
-    #     e = load('elements')
-    #     self.assertEqual(118, e.shape[0])
-    #     self.assertEqual(74, e.shape[1])
-    #     e = load('mp_inorganic')
-    #     self.assertEqual(69640, e.shape[0])
-    #     self.assertEqual(15, e.shape[1])
-    #     e = load('electron_density')
-    #     self.assertEqual(103, e.shape[0])
-    #     self.assertEqual(401, e.shape[1])
-
-    def test_pass(self):
-        self.assertEqual(True, True, 'should pass')
+def test_pass(setup):
+    print(setup)
+    assert 1, 'should pass'

travis/install.sh

@@ -22,6 +22,8 @@ else
 fi
 
 source activate xenonpy
+conda install pytest pytest-cov pylint
+
 
 python --version
 python -c "import numpy; print('numpy %s' % numpy.__version__)"