From 665985ec6dab76a873fd87a12f15e758817acc9b Mon Sep 17 00:00:00 2001
From: Shlomi Fish <shlomif@shlomifish.org>
Date: Tue, 1 May 2018 17:09:12 +0300
Subject: [PATCH] Add python2 tests.

---
 .../{unit/import_file.py => lib/pysol_tests/import_file1.py} | 2 +-
 tests/unit/import_file1.py                                   | 5 +++++
 tests/unit/import_file1_py3.py                               | 5 +++++
 3 files changed, 11 insertions(+), 1 deletion(-)
 rename tests/{unit/import_file.py => lib/pysol_tests/import_file1.py} (99%)
 create mode 100644 tests/unit/import_file1.py
 create mode 100644 tests/unit/import_file1_py3.py

diff --git a/tests/unit/import_file.py b/tests/lib/pysol_tests/import_file1.py
similarity index 99%
rename from tests/unit/import_file.py
rename to tests/lib/pysol_tests/import_file1.py
index 60f2dee6..882e37a9 100644
--- a/tests/unit/import_file.py
+++ b/tests/lib/pysol_tests/import_file1.py
@@ -123,7 +123,7 @@ KD QC 5C QH 6S 3D
         self.assertEqual(1, 1, 'card2str2 works')
 
 
-if __name__ == '__main__':
+def mymain():
     from pycotap import TAPTestRunner
     suite = unittest.TestLoader().loadTestsFromTestCase(MyTests)
     TAPTestRunner().run(suite)
diff --git a/tests/unit/import_file1.py b/tests/unit/import_file1.py
new file mode 100644
index 00000000..08ec28c6
--- /dev/null
+++ b/tests/unit/import_file1.py
@@ -0,0 +1,5 @@
+#!/usr/bin/python
+
+from pysol_tests.import_file1 import mymain
+
+mymain()
diff --git a/tests/unit/import_file1_py3.py b/tests/unit/import_file1_py3.py
new file mode 100644
index 00000000..c7031e81
--- /dev/null
+++ b/tests/unit/import_file1_py3.py
@@ -0,0 +1,5 @@
+#!/usr/bin/python3
+
+from pysol_tests.import_file1 import mymain
+
+mymain()