Added test

alishobeiri · alishobeiri · commit f700c92ca052 · 2017-06-27T17:02:54.000-04:00
diff --git a/plotly/tests/test_optional/test_figure_factory.py b/plotly/tests/test_optional/test_figure_factory.py
@@ -1999,6 +1999,15 @@ def test_valid_color_dict(self):
                                 data, 'a', 'b', color_name='a',
                                 colormap=color_dict)
 
+
+    def test_valid_index(self):
+        data = pd.DataFrame([[2, 3, 4], [4, 5, 6]],
+                            columns=['a', 'b', 'c'])
+
+        self.assertRaises(KeyError, ff.create_facet_grid, data, 'a', 'b',
+                          color_name='c')
+
+
     def test_valid_colorscale_name(self):
         data = pd.DataFrame([[0, 1, 2], [3, 4, 5]],
                             columns=['a', 'b', 'c'])
