plotly
diff --git a/‎plotly/tests/test_optional/test_figure_factory.py
Copy file name to clipboardExpand all lines: plotly/tests/test_optional/test_figure_factory.py
+9Lines changed: 9 additions & 0 deletions b/‎plotly/tests/test_optional/test_figure_factory.py
Copy file name to clipboardExpand all lines: plotly/tests/test_optional/test_figure_factory.py
+9Lines changed: 9 additions & 0 deletions
@@ -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'])