def test_can_check_used_assignment(self): handler = CodeHandler() handler.varis = dict(a=0) handler.get_declared().clear() handler.add_declared("a", Block("a := 1")) block = Block('fmt.Println(a)') handler.scan_used([block]) self.assertEqual(len(list(handler.get_params())), 1) self.assertTrue("a" in handler.get_params())