Tweak setup.py for py.test (pytest?)

1 files changed, 2 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 63b1d76..bde8621 100755
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@ if sys.argv[-1] == 'test':
         print('py.test required.')
         sys.exit(1)
 
-    os.system('pytest test_tablib.py')
+    os.system('py.test test_tablib.py')
     sys.exit()
 
 setup(
@@ -70,4 +70,5 @@ setup(
         'Programming Language :: Python :: 3.1',
         'Programming Language :: Python :: 3.2',
     ),
+    tests_require=['pytest'],
 )