def test_should_return_only_frames_satisfy_predicate(self):
dataframe = create_dataframe(3)
batch = FrameBatch(frames=dataframe)
expression = type("AbstractExpression", (), {"evaluate": lambda x: [
False, False, True]})
plan = type("PPScanPlan", (), {"predicate": expression})
predicate_executor = PPExecutor(plan)
predicate_executor.append_child(DummyExecutor([batch]))
expected = batch[[2]]
filtered = list(predicate_executor.exec())[0]
self.assertEqual(expected, filtered)
def test_should_return_only_frames_satisfy_predicate(self):
frame_1 = Frame(1, np.ones((1, 1)), None)
frame_2 = Frame(1, 2 * np.ones((1, 1)), None)
frame_3 = Frame(1, 3 * np.ones((1, 1)), None)
batch = FrameBatch(frames=[
frame_1,
frame_2,
frame_3,
], info=None)
expression = type("AbstractExpression", (),
{"evaluate": lambda x: [False, False, True]})
plan = type("PPScanPlan", (), {"predicate": expression})
predicate_executor = PPExecutor(plan)
predicate_executor.append_child(DummyExecutor([batch]))
expected = FrameBatch(frames=[frame_3], info=None)
filtered = list(predicate_executor.exec())[0]
self.assertEqual(expected, filtered)