From 3d32f57c6ff3157143a089de096a5ef7b237a706 Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Tue, 17 Oct 2017 15:13:30 -0500
Subject: [PATCH] Add coverage to unit tests

---
 .travis.yml | 2 +-
 setup.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 0de30b20a4..b4462ed0e7 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -62,6 +62,6 @@ script:
       ./check_source.py;
     else
       ./run_tests.py -C $OPENMC_CONFIG -j 2;
-      pytest unit_tests/;
+      pytest --cov=../openmc unit_tests/;
     fi
   - cd ..
diff --git a/setup.py b/setup.py
index e064efa68c..271fac0f00 100755
--- a/setup.py
+++ b/setup.py
@@ -66,7 +66,7 @@ kwargs = {
 
     # Optional dependencies
     'extras_require': {
-        'test': ['pytest'],
+        'test': ['pytest', 'pytest-cov'],
         'vtk': ['vtk', 'silomesh'],
     },
 }