Ejemplo n.º 1
0
    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)
Ejemplo n.º 2
0
    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())
Ejemplo n.º 3
0
    def test_get_field_count():
        schema = TableSchema(
            ["a", "b", "c"],
            [DataTypes.INT(),
             DataTypes.BIGINT(),
             DataTypes.STRING()])

        count = schema.get_field_count()

        assert count == 3
Ejemplo n.º 4
0
    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()
        ]