def test_visit_int_array_literal(self):
''' Testing int array literal
Function: visitArrayLiteral
'''
ctx = MagicMock()
visitor = ParserVisitor()
ctx.getText.return_value = '[1,2,3,4]'
expected = visitor.visitArrayLiteral(ctx)
self.assertEqual(expected.evaluate(),
Batch(pd.DataFrame({0: [np.array([1, 2, 3, 4])]})))
def test_visit_str_array_literal(self):
''' Testing str array literal
Function: visitArrayLiteral
'''
ctx = MagicMock()
visitor = ParserVisitor()
ctx.getText.return_value = "['person', 'car']"
expected = visitor.visitArrayLiteral(ctx)
self.assertEqual(
expected.evaluate(),
Batch(pd.DataFrame({0: [np.array(['person', 'car'])]})))