def test_get_field_count(self): schema = TableSchema(["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) count = schema.get_field_count() self.assertEqual(3, count)
def test_init(self): schema = TableSchema(["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) self.assertEqual(3, schema.get_field_count()) self.assertEqual(["a", "b", "c"], schema.get_field_names()) self.assertEqual([DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()], schema.get_field_data_types())
def test_get_field_count(): schema = TableSchema( ["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) count = schema.get_field_count() assert count == 3
def test_init(self): schema = TableSchema( ["a", "b", "c"], [DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING()]) assert schema.get_field_count() == 3 assert schema.get_field_names() == ["a", "b", "c"] assert schema.get_field_data_types() == [ DataTypes.INT(), DataTypes.BIGINT(), DataTypes.STRING() ]