def test_class(self): template = m.TemplateClass("ClassTpl") template.add_instantiation(int, DummyA) template.add_instantiation(float, DummyB) self.assertEquals(template[int], DummyA) self.assertEquals(str(DummyA), "<class '__main__.ClassTpl[int]'>") self.assertEquals(template[float], DummyB) self.assertEquals(str(DummyB), "<class '__main__.ClassTpl[float]'>")
def test_class(self): template = m.TemplateClass("ClassTpl") self.assertEquals(str(template), "<TemplateClass {}.ClassTpl>".format(_TEST_MODULE)) template.add_instantiation(int, DummyA) template.add_instantiation(float, DummyB) self.assertEquals(template[int], DummyA) self.assertEquals(str(DummyA), "<class '{}.ClassTpl[int]'>".format(_TEST_MODULE)) self.assertEquals(template[float], DummyB) self.assertEquals(str(DummyB), "<class '{}.ClassTpl[float]'>".format(_TEST_MODULE))