Adding tests for failing sqllab data type

apache · Mar 16, 2018 · 12caf65 · 12caf65
1 parent cf8a72b
commit 12caf65
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/sqllab_tests.py b/tests/sqllab_tests.py
@@ -210,6 +210,14 @@ def test_df_conversion_no_dict(self):
         self.assertEquals(len(data), cdf.size)
         self.assertEquals(len(cols), len(cdf.columns))
 
+    def test_df_conversion_tuple(self):
+        cols = [['string_col'], ['int_col'], ['list_col'], ['float_col']]
+        data = [(u'Text', 111, [123], 1.0)]
+        cdf = convert_results_to_df(cols, data)
+
+        self.assertEquals(len(data), cdf.size)
+        self.assertEquals(len(cols), len(cdf.columns))
+
     def test_df_conversion_dict(self):
         cols = [['string_col'], ['dict_col'], ['int_col']]
         data = [['a', {'c1': 1, 'c2': 2, 'c3': 3}, 4]]