Sankey charts tests

Author: WhiteBlackGoose

src/Plotly.NET/Playground.fsx

@@ -802,18 +802,3 @@ let values,labels =
     |> Seq.unzip
 
 let cols =[|"black";"blue"|]
-
-let r    = [77.5; 72.5; 70.0; 45.0; 22.5; 42.5; 40.0; 62.5]
-let r'   = [57.5; 50.0; 45.0; 35.0; 20.0; 22.5; 37.5; 55.0]
-let r''  = [40.0; 30.0; 30.0; 35.0; 7.5; 7.5; 32.5; 40.0]
-let r''' = [20.0; 7.5; 15.0; 22.5; 2.5; 2.5; 12.5; 22.5]
-
-let t = ["North"; "N-E"; "East"; "S-E"; "South"; "S-W"; "West"; "N-W"]
-[
-    Chart.WindRose (r   ,t,Name="11-14 m/s")
-    Chart.WindRose (r'  ,t,Name="8-11 m/s")
-    Chart.WindRose (r'' ,t,Name="5-8 m/s")
-    Chart.WindRose (r''',t,Name="< 5 m/s")
-]
-|> Chart.Combine
-|> Chart.Show

tests/Plotly.NET.Tests/HtmlCodegen/CategoricalCharts.fs

@@ -0,0 +1,117 @@
+module Tests.CategoricalCharts
+
+open Expecto
+open Plotly.NET
+open TestUtils
+open Plotly.NET.GenericChart
+open System
+
+let parallelCategoriesChart =
+    let dims =
+        [
+            Dimensions.init(["Cat1";"Cat1";"Cat1";"Cat1";"Cat2";"Cat2";"Cat3"],Label="A")
+            Dimensions.init([0;1;0;1;0;0;0],Label="B",TickText=["YES";"NO"])
+        ]
+    
+    Chart.ParallelCategories(
+        dims,
+        Color=[0.;1.;0.;1.;0.;0.;0.],
+        Colorscale = StyleParam.Colorscale.Blackbody
+    )
+
+[<Tests>]
+let ``Parallel categories charts`` =
+    testList "CategoricalCharts.Parallel categories charts" [
+        testCase "Parallel categories data" ( fun () ->
+            "var data = [{\"type\":\"parcats\",\"dimensions\":[{\"values\":[\"Cat1\",\"Cat1\",\"Cat1\",\"Cat1\",\"Cat2\",\"Cat2\",\"Cat3\"],\"label\":\"A\"},{\"values\":[0,1,0,1,0,0,0],\"label\":\"B\",\"ticktext\":[\"YES\",\"NO\"]}],\"color\":[0.0,1.0,0.0,1.0,0.0,0.0,0.0],\"line\":{\"colorscale\":\"Blackbody\"}}];"
+            |> chartGeneratedContains parallelCategoriesChart
+        );
+        testCase "Parallel categories layout" ( fun () ->
+            emptyLayout parallelCategoriesChart
+        );
+    ]
+
+
+let parcoords1Chart =
+    let data = 
+        [
+            "A",[1.;4.;3.4;0.7;]
+            "B",[3.;1.5;1.7;2.3;]
+            "C",[2.;4.;3.1;5.]
+            "D",[4.;2.;2.;4.;]
+        ]
+    Chart.ParallelCoord(data,Color="blue")
+
+let parcoordsChart = 
+    let v = [|
+        Dimensions.init([|1.;4.;|],  
+            StyleParam.Range.MinMax (1.,5.),StyleParam.Range.MinMax (1.,2.),Label="A");
+        Dimensions.init([|3.;1.5;|], 
+            StyleParam.Range.MinMax (1.,5.),Label="B",Tickvals=[|1.5;3.;4.;5.;|]);
+        Dimensions.init([|2.;4.;|],  
+            StyleParam.Range.MinMax (1.,5.),Label="C",Tickvals=[|1.;2.;4.;5.;|],
+                TickText=[|"txt 1";"txt 2";"txt 4";"txt 5";|]);
+        Dimensions.init([|4.;2.;|],  
+            StyleParam.Range.MinMax (1.,5.),Label="D");
+    |]
+
+    let dyn = Trace("parcoords")
+
+    dyn?dimensions <- v
+    dyn?line <- Line.init(Color = "blue")
+
+    dyn
+    |> GenericChart.ofTraceObject
+
+[<Tests>]
+let ``Parallel coordinates charts`` =
+    testList "CategoricalCharts.Parallel coordinates charts" [
+        testCase "Parallel coordinates 1 data" ( fun () ->
+            "var data = [{\"type\":\"parcoords\",\"dimensions\":[{\"values\":[1.0,4.0,3.4,0.7],\"label\":\"A\"},{\"values\":[3.0,1.5,1.7,2.3],\"label\":\"B\"},{\"values\":[2.0,4.0,3.1,5.0],\"label\":\"C\"},{\"values\":[4.0,2.0,2.0,4.0],\"label\":\"D\"}],\"line\":{\"color\":\"blue\"}}];"
+            |> chartGeneratedContains parcoords1Chart
+        );
+        testCase "Parallel coordinates 1 layout" ( fun () ->
+            emptyLayout parcoords1Chart
+        );
+        testCase "Parallel coordinates data" ( fun () ->
+            "var data = [{\"type\":\"parcoords\",\"dimensions\":[{\"values\":[1.0,4.0],\"range\":[1.0,5.0],\"constraintrange\":[1.0,2.0],\"label\":\"A\"},{\"values\":[3.0,1.5],\"range\":[1.0,5.0],\"label\":\"B\",\"tickvals\":[1.5,3.0,4.0,5.0]},{\"values\":[2.0,4.0],\"range\":[1.0,5.0],\"label\":\"C\",\"tickvals\":[1.0,2.0,4.0,5.0],\"ticktext\":[\"txt 1\",\"txt 2\",\"txt 4\",\"txt 5\"]},{\"values\":[4.0,2.0],\"range\":[1.0,5.0],\"label\":\"D\"}],\"line\":{\"color\":\"blue\"}}];"
+            |> chartGeneratedContains parcoordsChart
+        );
+        testCase "Parallel coordinates layout" ( fun () ->
+            emptyLayout parcoordsChart
+        );
+    ]
+
+
+let sankey1 = 
+    // create nodes
+    let n1 = Node.Create("a",color="Black")
+    let n2 = Node.Create("b",color="Red")
+    let n3 = Node.Create("c",color="Purple")
+    let n4 = Node.Create("d",color="Green")
+    let n5 = Node.Create("e",color="Orange")
+    
+    // create links between nodes
+    let link1 = Link.Create(n1,n2,value=1.0)
+    let link2 = Link.Create(n2,n3,value=2.0)
+    let link3 = Link.Create(n1,n5,value=1.3)
+    let link4 = Link.Create(n4,n5,value=1.5)
+    let link5 = Link.Create(n3,n5,value=0.5)
+    Chart.Sankey(
+        [n1;n2;n3;n4;n5],
+        [link1;link2;link3;link4;link5]
+    )
+    |> Chart.withTitle "Sankey Sample"
+
+[<Tests>]
+let ``Sankey charts`` =
+    testList "CategoricalCharts.Sankey charts" [
+        testCase "Sankey data" ( fun () ->
+            "var data = [{\"type\":\"sankey\",\"node\":{\"label\":[\"a\",\"b\",\"c\",\"d\",\"e\"],\"color\":[\"Black\",\"Red\",\"Purple\",\"Green\",\"Orange\"]},\"link\":{\"source\":[0,1,0,3,2],\"target\":[1,2,4,4,4],\"value\":[1.0,2.0,1.3,1.5,0.5]}}];"
+            |> chartGeneratedContains sankey1
+        );
+        testCase "Sankey layout" ( fun () ->
+            "var layout = {\"title\":\"Sankey Sample\"};"
+            |> chartGeneratedContains sankey1
+        );
+    ]

tests/Plotly.NET.Tests/Plotly.NET.Tests.fsproj

@@ -17,6 +17,7 @@
     <Compile Include="HtmlCodegen\MapboxMapCharts.fs" />
     <Compile Include="HtmlCodegen\FinanceCharts.fs" />
     <Compile Include="HtmlCodegen\PolarCharts.fs" />
+    <Compile Include="HtmlCodegen\CategoricalCharts.fs" />
     <Compile Include="ImageExport.fs" />
     <Compile Include="Main.fs" />
   </ItemGroup>