def test_get_field_name(self): schema = TableSchema(["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) field_name = schema.get_field_name(2) field_name_not_exist = schema.get_field_name(3) self.assertEqual("c", field_name) self.assertIsNone(field_name_not_exist)
def test_get_field_name(self): schema = TableSchema( ["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) field_name = schema.get_field_name(2) field_name_not_exist = schema.get_field_name(3) assert field_name == "c" assert field_name_not_exist is None