def test_builder(self): schema_builder = TableSchema.builder() schema = schema_builder \ .field("a", DataTypes.INT())\ .field("b", DataTypes.BIGINT())\ .field("c", DataTypes.STRING()).build() expected = TableSchema(["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) self.assertEqual(expected, schema)
def test_builder(): schema_builder = TableSchema.builder() schema = schema_builder \ .field("a", DataTypes.INT())\ .field("b", DataTypes.BIGINT())\ .field("c", DataTypes.STRING()).build() expected = TableSchema( ["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) assert schema == expected