Merge pull request #34 from kpj/feature-test

Add test including both seaborn and plotnine

tests/test_patchworklib.py

@@ -1,5 +1,6 @@
 from pathlib import Path
 
+import plotnine as p9
 import seaborn as sns
 
 import patchworklib as pw
@@ -25,3 +26,26 @@ def test_example_plot(tmp_path: Path):
  ax12 = ax1 | ax2
  ax12.savefig(result_file)
  assert result_file.exists()
+
+
+def test_sns_and_p9(tmp_path: Path):
+ titanic = sns.load_dataset("titanic")
+
+ g_sns = pw.Brick(figsize=(4, 4))
+ sns.boxplot(data=titanic, x="sex", y="survived", hue="class", ax=g_sns)
+ g_sns.set_title("seaborn")
+
+ g_p9 = pw.load_ggplot(
+ (
+ p9.ggplot(titanic, p9.aes(x="sex", y="survived", fill="class"))
+ + p9.geom_boxplot()
+ + p9.ggtitle("plotnine")
+ ),
+ figsize=(4, 4),
+ )
+
+ g = g_sns | g_p9
+
+ result_file = tmp_path / "g.png"
+ g.savefig(result_file)
+ assert result_file.exists()