def test_map_omit_missing_table_columns(self): # Currently, "omit" means "set empty". There's a valid use case for # actually _removing_ colnames here, but [adamhooper, 2019-01-04] we # haven't defined that case yet. dtype = ParamDType.Map(value_dtype=ParamDType.Column()) value = dtype.omit_missing_table_columns({'a': 'X', 'b': 'Y'}, {'X'}) self.assertEqual(value, {'a': 'X', 'b': ''})
def test_clean_column_missing_becomes_empty_string(self): context = RenderContext( None, TableShape(3, [ Column('A', ColumnType.NUMBER), ]), None, None) schema = ParamDType.Dict({ 'column': ParamDType.Column(), }) value = {'column': 'B'} result = clean_value(schema, value, context) self.assertEqual(result, {'column': ''})