def test_get(self): smap = SymbolMap() labeler = TextLabeler() self.assertEqual('x', smap.getSymbol(self.instance.x, labeler)) self.assertEqual('y(1)', smap.getSymbol(self.instance.y[1], labeler)) self.assertEqual(set(smap.bySymbol.keys()), set(['x', 'y(1)'])) self.assertEqual('x', smap.getSymbol(self.instance.x, labeler))
def test_error1(self): smap = SymbolMap() labeler = TextLabeler() self.assertEqual('x', smap.getSymbol(self.instance.x, labeler)) class FOO(object): def __call__(self, *args): return 'x' labeler = FOO() try: self.assertEqual('x', smap.getSymbol(self.instance.y[1], labeler)) self.fail("Expected RuntimeError") except RuntimeError: pass