Пример #1
0
    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)
Пример #2
0
    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