def test_load_data_statement(self):
parser = Parser()
load_data_query = """LOAD DATA INFILE 'data/video.mp4' INTO MyVideo;"""
expected_stmt = LoadDataStatement(TableRef(TableInfo('MyVideo')),
Path('data/video.mp4'))
eva_statement_list = parser.parse(load_data_query)
self.assertIsInstance(eva_statement_list, list)
self.assertEqual(len(eva_statement_list), 1)
self.assertEqual(eva_statement_list[0].stmt_type,
StatementType.LOAD_DATA)
load_data_stmt = eva_statement_list[0]
self.assertEqual(load_data_stmt, expected_stmt)
def test_should_return_false_for_unequal_expression(self):
table = TableRef(TableInfo('MyVideo'))
load_stmt = LoadDataStatement(table, Path('data/video.mp4'))
insert_stmt = InsertTableStatement(table)
create_udf = CreateUDFStatement('udf', False, [
ColumnDefinition('frame', ParserColumnDataType.NDARRAY,
[3, 256, 256])
], [ColumnDefinition('labels', ParserColumnDataType.NDARRAY, [10])],
Path('data/fastrcnn.py'),
'Classification')
select_stmt = SelectStatement()
self.assertNotEqual(load_stmt, insert_stmt)
self.assertNotEqual(insert_stmt, load_stmt)
self.assertNotEqual(create_udf, insert_stmt)
self.assertNotEqual(select_stmt, create_udf)
def visitLoadStatement(self, ctx: evaql_parser.LoadStatementContext):
file_path = self.visit(ctx.fileName()).value
table = self.visit(ctx.tableName())
stmt = LoadDataStatement(table, file_path)
return stmt