try some new tests

AUMAG · Mar 10, 2024 · 8976a14 · 8976a14
1 parent 4893367
commit 8976a14
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pysankey/tests/test_fruit_defaults.py b/pysankey/tests/test_fruit_defaults.py
@@ -5,7 +5,23 @@
 class TestFruitDefault(TestFruit):
 
     def test_fruits_default(self):
-        
+
         plt.figure(dpi=150)
         sky.sankey(self.data)
         plt.savefig("fruits_default.png")
+
+    def test_fruits_sorting(self):
+
+        plt.figure(dpi=150)
+        sky.sankey(self.data, sorting=1)
+        plt.savefig("fruits_sorting_1.png")
+
+        plt.figure(dpi=150)
+        sky.sankey(self.data, sorting=-1)
+        plt.savefig("fruits_sorting_2.png")
+
+    def test_fruits_colormap(self):
+
+        plt.figure(dpi=150)
+        sky.sankey(self.data, colormap="jet")
+        plt.savefig("fruits_colormap_jet.png")