def test_integer_list_constant(self):
input = [1, 2, 3]
node = ast.parse(str(input)).body[0].value
expected_output = ListType(Type.int)
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_string_tuple_constant(self):
input = (1, 2, 3)
node = ast.parse(str(input)).body[0].value
expected_output = TupleType(Type.int)
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_any_tuple_constant(self):
input = (1, '2', False)
node = ast.parse(str(input)).body[0].value
expected_output = TupleType(UnionType.build([Type.str, Type.int]))
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_boolean_tuple_constant(self):
input = (True, False)
node = ast.parse(str(input)).body[0].value
expected_output = TupleType(Type.bool)
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_string_constant(self):
input = 'unit_test'
node = ast.parse(str(input)).body[0].value
expected_output = Type.str
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_negative_integer_constant(self):
input = -10
node = ast.parse(str(input)).body[0].value
expected_output = Type.int
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def test_any_list_constant(self):
input = [1, '2', False]
node = ast.parse(str(input)).body[0].value
expected_output = ListType(Type.any)
typeanalyser = TypeAnalyser(Analyser(node), {})
output = typeanalyser.get_type(input)
self.assertEqual(expected_output, output)
def __check_types(self) -> bool:
"""
Performs the type checking
:return: a boolean value that represents if the analysis was successful
"""
type_analyser = TypeAnalyser(self, self.symbol_table, log=self._log)
return not type_analyser.has_errors
