def test_scan(self): with HTTMock(create_mock()): relation = MyriaRelation(FULL_NAME, connection=self.connection) json = relation._sink().to_json() optype = json['plan']['fragments'][0]['operators'][0]['opType'] name = json['plan']['fragments'][0]['operators'][0]['opName'] self.assertTrue('Scan' in optype) self.assertTrue(relation.name in name)
def test_load(self): state = {} with HTTMock(create_mock(state)): url = 'file:///foo.bar' relation = MyriaRelation(FULL_NAME, connection=self.connection) relation.load(url, schema=MyriaSchema(SCHEMA)) plan = relation._sink().to_json() text = json.dumps(plan['plan']['fragments'][1]['operators'][0]) self.assertTrue('FileScan' in text) self.assertTrue('TupleSource' in text) self.assertTrue(url in text) relation.execute() plan = state['query'] text = json.dumps(plan['plan']['fragments'][0]['operators'][0]) self.assertTrue('FileScan' in text) self.assertTrue('TupleSource' in text) self.assertTrue(url in text)