Allow cimport for quicktions

test_quicktions.py

@@ -1,49 +1,48 @@
 import os
+import glob
 import unittest
 
+from Cython.Build import Cythonize
+
 import quicktions
 F = quicktions.Fraction
 gcd = quicktions._gcd
 
 class CImportTest(unittest.TestCase):
 
     def setUp(self):
-        self.build_test_module()
+        self.module_files = []
 
     def tearDown(self):
-        self.remove_test_module()
+        for fn in self.module_files:
+            if os.path.exists(fn):
+                os.remove(fn)
 
     def build_test_module(self):
-        self.module_code = '\n'.join([
+        module_code = '\n'.join([
             '# cython: language_level=3str',
             'from quicktions cimport Fraction',
             'def get_fraction():',
             '    return Fraction(1, 2)',
         ])
-        self.base_path = os.path.abspath(os.path.dirname(__file__))
-        self.module_name = 'quicktions_importtest'
-        self.module_filename = os.path.join(self.base_path, '.'.join([self.module_name, 'pyx']))
-        with open(self.module_filename, 'w') as f:
-            f.write(self.module_code)
-
-    def remove_test_module(self):
-        for fn in os.listdir(self.base_path):
-            if not fn.startswith(self.module_name):
-                continue
-            os.remove(os.path.join(self.base_path, fn))
+        base_path = os.path.abspath(os.path.dirname(__file__))
+        module_name = 'quicktions_importtest'
+        module_filename = os.path.join(base_path, '.'.join([module_name, 'pyx']))
+        with open(module_filename, 'w') as f:
+            f.write(module_code)
+
+        Cythonize.main(['-i', module_filename])
+
+        for fn in glob.glob(os.path.join(base_path, '.'.join([module_name, '*']))):
+            self.module_files.append(os.path.abspath(fn))
 
     def test_cimport(self):
         self.build_test_module()
-        import pyximport
-        self.py_importer, self.pyx_importer = pyximport.install(inplace=True, language_level=3)
 
         from quicktions_importtest import get_fraction
 
         self.assertEqual(get_fraction(), F(1,2))
 
-        pyximport.uninstall(self.py_importer, self.pyx_importer)
-
-
 
 def test_main():
     suite = unittest.TestSuite()