def test_op_type_check_with_wrong_type(): """ with input type TYPE_IDENTIFIER, check if None is returned by op_type_check. """ meta = TM({ "tables": [ {"fields": [{'name': 'col', 'type': TM.TYPE_IDENTIFIER}]} ]}) op = FakeOp('col') meta2 = op.op_type_check(meta) assert meta2 is None assert meta.get_type('col') == TM.TYPE_IDENTIFIER assert op.input_type is TM.TYPE_IDENTIFIER and op.output_type is None
def test_op_type_check_with_correct_type2(): """ With input type TYPE_TEXT, check if input_type and output_type are correct. """ meta = TM({ "tables": [ {"fields": [{'name': 'col', 'type': TM.TYPE_TEXT}]} ]}) op = FakeOp('col') meta2 = op.op_type_check(meta) assert meta2 == meta assert meta.get_type('col') == TM.TYPE_BOOL assert op.input_type == TM.TYPE_TEXT and op.output_type == TM.TYPE_BOOL