예제 #1
0
    def test_schema_from_dframe_cardnalities(self):
        schema = schema_from_dframe(self.dframe)

        for column, column_schema in schema.items():
            card = column_schema[CARDINALITY]
            self.assertTrue(card <= len(self.dframe))
            self.assertTrue(card >= 0)

            if card == 0:
                self.assertTrue(all([isnan(x) for x in self.dframe[column]]))
예제 #2
0
    def test_schema_from_dframe_unique_encoded_columns(self):
        self.dframe.rename(columns={
            'food_type': 'rating+',
            'comments': 'rating-'
        },
                           inplace=True)
        schema = schema_from_dframe(self.dframe)

        self.assertTrue('rating_' in schema)
        self.assertTrue('rating__' in schema)
예제 #3
0
    def test_schema_from_dframe_no_reserved_keys(self):
        for key in RESERVED_KEYS:
            self.dframe[key] = 1

        for key in RESERVED_KEYS:
            self.assertTrue(key in self.dframe.columns)

        schema = schema_from_dframe(self.dframe)

        for key in RESERVED_KEYS:
            self.assertFalse(key in schema)