Merge pull request #322 from BibMartin/colormap_tests

Add test_colormap.py

Author: ocefpaf

tests/test_colormap.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+""""
+Folium Colormap Module
+----------------------
+"""
+import folium.colormap as cm
+
+def test_simple_step():
+    step = cm.StepColormap(['green','yellow','red'], vmin=3., vmax=10., index=[3,4,8,10], caption='step')
+    step = cm.StepColormap(['r','y','g','c','b','m'])
+    step._repr_html_()
+
+def test_simple_linear():
+    linear = cm.LinearColormap(['green','yellow','red'], vmin=3., vmax=10.)
+    linear = cm.LinearColormap(['red','orange', 'yellow','green'], index=[0,0.1,0.9,1.])
+    linear._repr_html_()
+
+def test_linear_to_step():
+    some_list = [30.6, 50, 51, 52, 53, 54, 55, 60, 70, 100]
+    lc = cm.linear.YlOrRd
+    lc.to_step(n=12)
+    lc.to_step(index=[0,2,4,6,8,10])
+    lc.to_step(data=some_list, n=12)
+    lc.to_step(data=some_list, n=12, method='linear')
+    lc.to_step(data=some_list, n=12, method='log')
+    lc.to_step(data=some_list, n=30, method='quantiles')
+    lc.to_step(data=some_list, quantiles=[0,0.3,0.7,1])
+    lc.to_step(data=some_list, quantiles=[0,0.3,0.7,1], round_method='int')
+    lc.to_step(data=some_list, quantiles=[0,0.3,0.7,1], round_method='log10')
+
+def test_step_to_linear():
+    step = cm.StepColormap(['green','yellow','red'], vmin=3., vmax=10., index=[3,4,8,10], caption='step')
+    step.to_linear()
+
+def test_linear_object():
+    cm.linear.OrRd._repr_html_()
+    cm.linear.PuBu.to_step(12)
+    cm.linear.YlGn.scale(3,12)
+    cm.linear._repr_html_()