pytest

.travis.yml

@@ -12,14 +12,14 @@ before_install:
     - bash miniconda.sh -b -p $HOME/miniconda
     - export PATH="$HOME/miniconda/bin:$PATH"
     - conda update --yes --all
-    - travis_retry conda create --yes -n test python=$PYTHON --file requirements.txt
-    - source activate test
+    - travis_retry conda create --yes -n TEST python=$PYTHON --file requirements.txt
+    - source activate TEST
     - travis_retry conda install --yes --file requirements-dev.txt
     - if [[ "$PYTHON" != "3.5" ]]; then
         travis_retry conda install --yes mock ;
       fi
     - travis_retry pip install
 
 script:
-    - nosetests
+    - python setup.py test -a "--verbose"
     - find . -type f -name "*.py" | xargs flake8 --max-line-length=100

requirements-dev.txt

@@ -1,3 +1,4 @@
 mock
 nose
+pytest
 flake8

requirements.txt

@@ -1,2 +1,2 @@
-netCDF4 >=1.2.0
+netCDF4 >=1.2.0,<1.2.2
 numpy

setup.py

@@ -2,7 +2,23 @@
                         with_statement)
 
 import os
+import sys
 from setuptools import setup, find_packages
+from setuptools.command.test import test as TestCommand
+
+
+class PyTest(TestCommand):
+    user_options = [('pytest-args=', 'a', "Arguments to pass to py.test")]
+
+    def initialize_options(self):
+        TestCommand.initialize_options(self)
+        self.pytest_args = []
+
+    def run_tests(self):
+        # Import here, cause outside the eggs aren't loaded.
+        import pytest
+        errno = pytest.main(self.pytest_args)
+        sys.exit(errno)
 
 
 def extract_version(module='pysgrid'):
@@ -36,7 +52,8 @@ def readme():
       license='BSD',
       long_description=readme(),
       install_requires=reqs,
-      tests_require=['mock', 'nose'],
+      tests_require=['mock', 'nose', 'pytest'],
+      cmdclass=dict(test=PyTest),
       classifiers=[
           'Development Status :: 3 - Alpha',
           'Intended Audience :: Developers',
@@ -47,4 +64,5 @@ def readme():
           'Topic :: Scientific/Engineering',
           ],
       include_package_data=True,
+      zip_safe=False
       )