fix: types in list

widgetti · Aug 2, 2023 · f270d6e · f270d6e
1 parent a3ff311
commit f270d6e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/unit/slider_test.py b/tests/unit/slider_test.py
@@ -16,19 +16,19 @@ def get_tick_labels(clazz, min, max, step, **kwargs) -> list:
 def test_int_slider():
     reference_ticks = [3, 4, 5]
     assert get_tick_labels(solara.IntSlider, 3, 5, 1) == ["3", None, "5"]
-    assert get_tick_labels(solara.IntSlider, 3, 5, 1, tick_labels=None) == [None] * len(reference_ticks)
-    assert get_tick_labels(solara.IntSlider, 3, 5, 1, tick_labels=reference_ticks) == map(str, reference_ticks)
+    assert get_tick_labels(solara.IntSlider, 3, 5, 1, tick_labels=None) is None
+    assert get_tick_labels(solara.IntSlider, 3, 5, 1, tick_labels=reference_ticks) == reference_ticks
 
 
 def test_float_slider():
     reference_ticks = [3, 3.5, 4, 4.5, 5]
     middle_nones = [None] * (len(reference_ticks) - 2)
     assert get_tick_labels(solara.FloatSlider, 3, 5, 0.5) == ["3", *middle_nones, "5"]
-    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.5, tick_labels=None) == [None] * len(reference_ticks)
-    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.5, tick_labels=reference_ticks) == map(str, reference_ticks)
+    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.5, tick_labels=None) is None
+    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.5, tick_labels=reference_ticks) == reference_ticks
 
     reference_ticks = [3, 3.3, 3.6, 3.9, 4.2, 4.5, 4.8, 5]
     middle_nones = [None] * (len(reference_ticks) - 2)
     assert get_tick_labels(solara.FloatSlider, 3, 5, 0.3) == ["3", *middle_nones, "5"]
-    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.3, tick_labels=None) == [None] * len(reference_ticks)
-    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.3, tick_labels=reference_ticks) == map(str, reference_ticks)
+    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.3, tick_labels=None) is None
+    assert get_tick_labels(solara.FloatSlider, 3, 5, 0.3, tick_labels=reference_ticks) == reference_ticks