def testInsertionAndRetrieval(self):
table = SymbolTable()
inttype = TypeInfo(rvalue=False, baseType="int")
floattype = TypeInfo(rvalue=False, baseType="float")
chartype = TypeInfo(rvalue=False, baseType="char")
a = ASTVariableNode("a")
a.typeInfo = inttype
b = ASTVariableNode("b")
b.typeInfo = floattype
c = ASTVariableNode("c")
c.typeInfo = chartype
d = ASTVariableNode("d")
d.typeInfo = floattype
b_bis = ASTVariableNode("b")
b_bis.typeInfo = inttype
# d_bis = ASTVariableNode("d")
# d_bis.type = floattype
table.insertVariableSymbol(a)
table.insertVariableSymbol(b)
table.openScope()
table.insertVariableSymbol(c)
self.assertTrue(table.retrieveSymbol("a", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("b", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("c", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("d", requireSeen=False) is None)
table.closeScope()
table.openScope()
table.insertVariableSymbol(d)
table.insertVariableSymbol(b_bis)
self.assertTrue(table.retrieveSymbol("a", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("b", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("c", requireSeen=False) is None)
self.assertTrue(table.retrieveSymbol("d", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("b", requireSeen=False).typeInfo.baseType == "int")
table.closeScope()
self.assertTrue(table.retrieveSymbol("b", requireSeen=False).typeInfo.baseType == "float")
self.assertTrue(table.retrieveSymbol("a", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("b", requireSeen=False) is not None)
self.assertTrue(table.retrieveSymbol("c", requireSeen=False) is None)
self.assertTrue(table.retrieveSymbol("d", requireSeen=False) is None)
