From 178dcc8bc4a277f58350f43c139464a6e7b59656 Mon Sep 17 00:00:00 2001
From: Henry Fredrick Schreiner <henry.fredrick.schreiner@cern.ch>
Date: Fri, 1 Mar 2019 13:30:28 +0100
Subject: [PATCH] Adding tests require

---
 .travis.yml | 8 +++++++-
 setup.py    | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/.travis.yml b/.travis.yml
index 7e8ab0b..1b841af 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,5 +1,9 @@
 language: python
 
+branches:
+  only:
+    - master
+
 os:
   - linux
 
@@ -18,10 +22,12 @@ addons:
 
 install:
   - pip install --upgrade pip
-  - pip install coveralls pytest pytest-cov
+  - pip install --upgrade pytest
+  - pip install coveralls pytest-cov
 
 before_script:
   pip install -e .
+
 script:
   pytest --cov=numpythia tests/
 
diff --git a/setup.py b/setup.py
index 8ad72a7..1cd84be 100755
--- a/setup.py
+++ b/setup.py
@@ -148,5 +148,6 @@ def finalize_options(self):
         'Development Status :: 4 - Beta',
     ],
     install_requires=['numpy', 'six'],
+    tests_require=['pytest'],
     zip_safe=False,
 )