def test_should_not_call_insert_generator_for_other_types( self, mock_class): PlanGenerator().build(Operator(None)) PlanGenerator().build(Operator(LogicalFilter(None))) PlanGenerator().build(Operator(LogicalGet(None, 1))) PlanGenerator().build(Operator(LogicalProject([]))) mock_class.assert_not_called()
def test_should_not_call_scan_generator_for_other_types(self, mock_class): PlanGenerator().build(Operator(None)) PlanGenerator().build(LogicalInsert(None, 1, [], [])) mock_class.assert_not_called()