Run Python 3 test script when running Py3

1 files changed, 7 insertions, 2 deletions

diff --git a/bin/test b/bin/test
index 0b7ce83..025248e 100755
--- a/bin/test
+++ b/bin/test
@@ -1,3 +1,8 @@
 #!/bin/bash
-nosetests --nocapture --with-doctest --doctest-tests -I ^test.py $* && \
-  pep8 --ignore=E402 screenplain tests
+if [[ $(python -V 2>&1) =~ "Python 2" ]]
+then
+    nosetests --nocapture --with-doctest --doctest-tests -I ^test.py $* && \
+      pep8 --ignore=E402 screenplain tests
+else
+    python test.py && pep8 --ignore=E402 screenplain tests
+fi