add tests for is_non_string_iterable

data-8 · Feb 9, 2023 · b5e2a8a · b5e2a8a
1 parent d122a21
commit b5e2a8a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -96,3 +96,14 @@ def test_proportions_from_distribution():
     uniform = u.column(1)
     assert len(uniform) == 50 and _round_eq(1, sum(uniform))
     assert [x in (0, 0.5, 1) for x in ds.sample_proportions(2, ds.make_array(.2, .3, .5))]
+
+
+def test_is_non_string_iterable():
+    is_string = 'hello'
+    assert ds.is_non_string_iterable(is_string) == False
+
+    is_list = [1, 2, 3]
+    assert ds.is_non_string_iterable(is_list) == True
+
+    is_int = 1
+    assert ds.is_non_string_iterable(is_int) == False