def test_00_template(self): # Compile template. temp = pdt.template(self.func) # Make sure template looks like the original function. self.assertTrue(inspect.isfunction(temp)) self.assertTrue(temp.__name__ == self.func.__name__) self.assertTrue(temp.__module__ == self.func.__module__) self.assertTrue(temp.__globals__ is self.func.__globals__) self.assertTrue(temp.__doc__ is None) # Make sure template output is what is expected. temp_str = temp(self.func_data) self.assertTrue(temp_str == self.func_str)
def test_01_template_with_args(self): # Get template decorator. dec = pdt.template() self.assertTrue(dec.__name__ == 'template_decorator') self.assertTrue(isinstance(dec.__closure__[0].cell_contents, pdt.Template)) # Compile template. temp = dec(self.func) # Make sure template looks like the original function. self.assertTrue(inspect.isfunction(temp)) self.assertTrue(temp.__name__ == self.func.__name__) self.assertTrue(temp.__module__ == self.func.__module__) self.assertTrue(temp.__globals__ is self.func.__globals__) self.assertTrue(temp.__doc__ is None) # Make sure template output is what is expected. temp_str = temp(self.func_data) self.assertTrue(temp_str == self.func_str)