Update test_vis.py (#315)

Issue when running test_vis in Github Actions. Adjusted test_scatter_chart and test_colored_scatter_chart
lux-org · Mar 20, 2021 · 127806f · 127806f
1 parent 182a4b7
commit 127806f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/tests/test_vis.py b/tests/test_vis.py
@@ -266,8 +266,11 @@ def test_scatter_chart(global_var):
     vis = Vis(["Acceleration", "Weight"], df)
     vis_code = vis.to_matplotlib_code()
     assert "ax.scatter(x_pts, y_pts, alpha=0.5)" in vis_code
-    assert "ax.set_xlabel('Acceleration')" in vis_code
-    assert "ax.set_ylabel('Weight')" in vis_code
+    assert (
+        "ax.set_xlabel('Acceleration', fontsize='15')" in vis_code
+        or "ax.set_xlabel('Acceleration')" in vis_code
+    )
+    assert "ax.set_ylabel('Weight', fontsize='15')" in vis_code or "ax.set_ylabel('Weight')" in vis_code
 
 
 def test_colored_scatter_chart(global_var):
@@ -290,8 +293,11 @@ def test_colored_scatter_chart(global_var):
     vis_code = vis.to_matplotlib_code()
     assert "ax.scatter" in vis_code
     assert "title='Origin'" in vis_code
-    assert "ax.set_xlabel('Acceleration')" in vis_code
-    assert "ax.set_ylabel('Weight')" in vis_code
+    assert (
+        "ax.set_xlabel('Acceleration', fontsize='15')" in vis_code
+        or "ax.set_xlabel('Acceleration')" in vis_code
+    )
+    assert "ax.set_ylabel('Weight', fontsize='15')" in vis_code or "ax.set_ylabel('Weight')" in vis_code
 
 
 def test_line_chart(global_var):