Fix tests

apache · Jun 27, 2018 · 437bbf4 · 437bbf4
1 parent 5127fdd
commit 437bbf4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/dataframe_test.py b/tests/dataframe_test.py
@@ -79,7 +79,8 @@ def test_get_columns_with_int(self):
                     'is_date': False,
                     'type': 'INT',
                     'name': 'b',
-                    'is_dim': True,
+                    'is_dim': False,
+                    'agg': 'sum',
                 },
             ],
         )
@@ -101,12 +102,14 @@ def test_get_columns_type_inference(self):
                     'is_date': False,
                     'type': 'FLOAT',
                     'name': 'a',
-                    'is_dim': True,
+                    'is_dim': False,
+                    'agg': 'sum',
                 }, {
                     'is_date': False,
                     'type': 'INT',
                     'name': 'b',
-                    'is_dim': True,
+                    'is_dim': False,
+                    'agg': 'sum',
                 },
             ],
         )