| Index: third_party/cython/src/Cython/Compiler/Tests/TestUtilityLoad.py
|
| diff --git a/third_party/cython/src/Cython/Compiler/Tests/TestUtilityLoad.py b/third_party/cython/src/Cython/Compiler/Tests/TestUtilityLoad.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..a7ec3ef8fe7ab6d7a89df52e561fef666f123d60
|
| --- /dev/null
|
| +++ b/third_party/cython/src/Cython/Compiler/Tests/TestUtilityLoad.py
|
| @@ -0,0 +1,101 @@
|
| +import unittest
|
| +
|
| +from Cython.Compiler import Code, UtilityCode
|
| +
|
| +
|
| +def strip_2tup(tup):
|
| + return tup[0] and tup[0].strip(), tup[1] and tup[1].strip()
|
| +
|
| +class TestUtilityLoader(unittest.TestCase):
|
| + """
|
| + Test loading UtilityCodes
|
| + """
|
| +
|
| + expected = "test {{loader}} prototype", "test {{loader}} impl"
|
| +
|
| + required = "req {{loader}} proto", "req {{loader}} impl"
|
| +
|
| + context = dict(loader='Loader')
|
| +
|
| + name = "TestUtilityLoader"
|
| + filename = "TestUtilityLoader.c"
|
| + cls = Code.UtilityCode
|
| +
|
| + def test_load_as_string(self):
|
| + got = strip_2tup(self.cls.load_as_string(self.name))
|
| + self.assertEquals(got, self.expected)
|
| +
|
| + got = strip_2tup(self.cls.load_as_string(self.name, self.filename))
|
| + self.assertEquals(got, self.expected)
|
| +
|
| + def test_load(self):
|
| + utility = self.cls.load(self.name)
|
| + got = strip_2tup((utility.proto, utility.impl))
|
| + self.assertEquals(got, self.expected)
|
| +
|
| + required, = utility.requires
|
| + got = strip_2tup((required.proto, required.impl))
|
| + self.assertEquals(got, self.required)
|
| +
|
| + utility = self.cls.load(self.name, from_file=self.filename)
|
| + got = strip_2tup((utility.proto, utility.impl))
|
| + self.assertEquals(got, self.expected)
|
| +
|
| + utility = self.cls.load_cached(self.name, from_file=self.filename)
|
| + got = strip_2tup((utility.proto, utility.impl))
|
| + self.assertEquals(got, self.expected)
|
| +
|
| +
|
| +class TestTempitaUtilityLoader(TestUtilityLoader):
|
| + """
|
| + Test loading UtilityCodes with Tempita substitution
|
| + """
|
| + expected_tempita = (TestUtilityLoader.expected[0].replace('{{loader}}', 'Loader'),
|
| + TestUtilityLoader.expected[1].replace('{{loader}}', 'Loader'))
|
| +
|
| + required_tempita = (TestUtilityLoader.required[0].replace('{{loader}}', 'Loader'),
|
| + TestUtilityLoader.required[1].replace('{{loader}}', 'Loader'))
|
| +
|
| + cls = Code.TempitaUtilityCode
|
| +
|
| + def test_load_as_string(self):
|
| + got = strip_2tup(self.cls.load_as_string(self.name, context=self.context))
|
| + self.assertEquals(got, self.expected_tempita)
|
| +
|
| + def test_load(self):
|
| + utility = self.cls.load(self.name, context=self.context)
|
| + got = strip_2tup((utility.proto, utility.impl))
|
| + self.assertEquals(got, self.expected_tempita)
|
| +
|
| + required, = utility.requires
|
| + got = strip_2tup((required.proto, required.impl))
|
| + self.assertEquals(got, self.required_tempita)
|
| +
|
| + utility = self.cls.load(self.name, from_file=self.filename, context=self.context)
|
| + got = strip_2tup((utility.proto, utility.impl))
|
| + self.assertEquals(got, self.expected_tempita)
|
| +
|
| +
|
| +class TestCythonUtilityLoader(TestTempitaUtilityLoader):
|
| + """
|
| + Test loading CythonUtilityCodes
|
| + """
|
| +
|
| + # Just change the attributes and run the same tests
|
| + expected = None, "test {{cy_loader}} impl"
|
| + expected_tempita = None, "test CyLoader impl"
|
| +
|
| + required = None, "req {{cy_loader}} impl"
|
| + required_tempita = None, "req CyLoader impl"
|
| +
|
| + context = dict(cy_loader='CyLoader')
|
| +
|
| + name = "TestCyUtilityLoader"
|
| + filename = "TestCyUtilityLoader.pyx"
|
| + cls = UtilityCode.CythonUtilityCode
|
| +
|
| + # Small hack to pass our tests above
|
| + cls.proto = None
|
| +
|
| + test_load = TestUtilityLoader.test_load
|
| + test_load_tempita = TestTempitaUtilityLoader.test_load
|
|
|