minor fixes

lux-org · Mar 22, 2021 · d9d39c2 · d9d39c2
1 parent f9d6b59
commit d9d39c2
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/tests/test_vis.py b/tests/test_vis.py
@@ -250,14 +250,10 @@ def test_colored_bar_chart(global_var):
 def test_bar_uniform():
     df = pd.read_csv("lux/data/car.csv")
     df["Year"] = pd.to_datetime(df["Year"], format="%Y")
-    df["Units"] = "units"
-    vis = Vis(["Units"], df)
+    df["Type"] = "A"
+    vis = Vis(["Type"], df)
     vis_code = vis.to_Altair()
-    assert (
-        "y = alt.Y('Units', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='Units'))"
-        in vis_code
-    )
-
+    assert ("y = alt.Y('Type', type= 'nominal'" in vis_code)
 
 def test_scatter_chart(global_var):
     df = pytest.car_df
@@ -373,16 +369,13 @@ def test_histogram_chart(global_var):
     assert "ax.set_ylabel('Number of Records')" in vis_code
 
 
-def test_uniform_histogram():
+def test_histogram_uniform():
     df = pd.read_csv("lux/data/car.csv")
     df["Year"] = pd.to_datetime(df["Year"], format="%Y")
     df["Units"] = 4.0
     vis = Vis(["Units"], df)
     vis_code = vis.to_Altair()
-    assert (
-        "y = alt.Y('Units', type= 'nominal', axis=alt.Axis(labelOverlap=True, title='Units'), sort ='-x'),"
-        in vis_code
-    )
+    assert ("y = alt.Y('Units', type= 'nominal'" in vis_code)
 
 
 def test_heatmap_chart(global_var):