adds test for colouring violin plots

TidierOrg · Apr 3, 2024 · 51e294d · 51e294d
1 parent eee32bc
commit 51e294d
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 1 deletion.
diff --git a/test/Manifest.toml b/test/Manifest.toml
@@ -2,7 +2,7 @@
 
 julia_version = "1.10.2"
 manifest_format = "2.0"
-project_hash = "34fb74b01d0db522e852db0c7efd7694431617fc"
+project_hash = "2c5b2138cb65ab82766fae5af59fccb44af83f7f"
 
 [[deps.AbstractFFTs]]
 deps = ["LinearAlgebra"]

diff --git a/test/Project.toml b/test/Project.toml
@@ -1,5 +1,6 @@
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
+Colors = "5ae59095-9a9b-59fe-a467-6f913c188581"
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 GLM = "38e38edf-8417-5370-95a0-9cbb8c7f171a"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -9,6 +9,7 @@ using CategoricalArrays
 using GLM
 using Loess
 using KernelDensity
+using Colors
 
 # functions to compare two images using a difference hash
 # essentially copied from ImageHashes.jl, but package is out of date

diff --git a/test/test_geoms.jl b/test/test_geoms.jl
@@ -135,6 +135,37 @@
         )
 
         @test plot_images_equal(t, m)
+
+        colours = [
+            RGB(0/255, 114/255, 178/255), # blue
+            RGB(230/255, 159/255, 0/255), # orange
+            RGB(0/255, 158/255, 115/255), # green
+            RGB(204/255, 121/255, 167/255), # reddish purple
+            RGB(86/255, 180/255, 233/255), # sky blue
+            RGB(213/255, 94/255, 0/255), # vermillion
+            RGB(240/255, 228/255, 66/255), # yellow
+        ]
+
+        m = Makie.plot(
+            Makie.SpecApi.GridLayout(
+                Makie.SpecApi.Axis(
+                    plots = [
+                        Makie.PlotSpec(
+                        :Violin,
+                        levelcode.(cat_array),
+                        penguins.bill_length_mm;
+                        color = (x -> colours[x]).(levelcode.(cat_array))
+                    )]; xticks = (unique(levelcode.(cat_array)),
+                    unique(cat_array))
+                )
+            )
+        )
+
+        t = ggplot(penguins) + 
+            geom_violin(aes(x = :species, y = :bill_length_mm, color = :species))
+
+        @test plot_images_equal(t, m)
+
     end
 
     @testset "geom_hist" begin