test before render

tdhopper · tdhopper · commit 1069f051244b · 2017-06-27T14:18:50.000-04:00
diff --git a/Makefile b/Makefile
@@ -11,9 +11,12 @@ clean:
 	rm -f *.pyc
 	rm -f  .Rhistory
 
-travis: render
+travis: test render
 	echo "Done"
 
+test:
+	python -m pytest tests/
+
 qrender:
 	python render.py "Examples.ipynb"
 
diff --git a/tests/test_plots.py b/tests/test_plots.py
@@ -0,0 +1,25 @@
+import render
+
+defined_plots = {'bar-counts': ['pandas', 'plotnine', 'ggplot'],
+                 'dodged-bar-chart': ['pandas', 'plotnine', 'ggplot'],
+                 'scatter-plot': ['pandas', 'plotnine', 'ggplot'],
+                 'scatter-plot-with-colors': ['matplotlib',
+                                              'seaborn',
+                                              'plotnine',
+                                              'ggplot'],
+                 'scatter-plot-with-facet': ['seaborn', 'plotnine', 'ggplot'],
+                 'scatter-plot-with-facets': ['seaborn', 'plotnine', 'ggplot'],
+                 'scatter-plot-with-size': ['pandas', 'plotnine', 'ggplot'],
+                 'scatter-with-regression': ['seaborn', 'plotnine', 'ggplot'],
+                 'simple-histogram': ['pandas', 'matplotlib', 'plotnine', 'ggplot', 'bokeh'],
+                 'stacked-bar-chart': ['pandas', 'plotnine', 'ggplot'],
+                 'stacked-kde': ['pandas', 'seaborn', 'plotnine', 'ggplot'],
+                 'stacked-smooth-line-and-scatter': ['plotnine', 'ggplot'],
+                 'timeseries': ['pandas', 'plotnine', 'ggplot']}
+
+
+def test_exist():
+    plots = render.extract_cells("Examples.ipynb")
+    for name, slug in plots:
+        names = [d['package-slug'] for d in plots[(name, slug)]]
+        assert set(names) == set(defined_plots[slug])
