diff --git a/tests/grids/test_core.py b/tests/grids/test_core.py index e52b0f63..67cbff6f 100644 --- a/tests/grids/test_core.py +++ b/tests/grids/test_core.py @@ -145,15 +145,17 @@ class TestGrid(WebTestCase): # can replace label grid.set_label('name', "Different") self.assertEqual(grid.labels['name'], "Different") - self.assertEqual(grid.get_label('name'), "Different") + self.assertEqual(grid.get_column_label('name'), "Different") + self.assertEqual(grid.get_filter_label('name'), "Different") # can update only column, not filter self.assertEqual(grid.labels, {'name': "Different"}) self.assertIn('name', grid.filters) self.assertEqual(grid.filters['name'].label, "Different") grid.set_label('name', "COLUMN ONLY", column_only=True) - self.assertEqual(grid.get_label('name'), "COLUMN ONLY") self.assertEqual(grid.filters['name'].label, "Different") + self.assertEqual(grid.get_column_label('name'), "COLUMN ONLY") + self.assertEqual(grid.get_filter_label('name'), "Different") def test_get_view_click_handler(self): model = self.app.model