def test_is_date(self): data = [("a", 1), ("a", 2)] cursor_descr = (("a", "string"), ("a", "string")) results = SupersetResultSet(data, cursor_descr, BaseEngineSpec) self.assertEqual(results.is_temporal("DATE"), True) self.assertEqual(results.is_temporal("DATETIME"), True) self.assertEqual(results.is_temporal("TIME"), True) self.assertEqual(results.is_temporal("TIMESTAMP"), True) self.assertEqual(results.is_temporal("STRING"), False) self.assertEqual(results.is_temporal(""), False) self.assertEqual(results.is_temporal(None), False)