tests: fix some tracking tests.

ghsanti · Oct 1, 2024 · 0c04af1 · 0c04af1
1 parent 769db03
commit 0c04af1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/keras_tuner/utils_test.py b/keras_tuner/utils_test.py
@@ -14,10 +14,11 @@
 
 
 from keras_tuner import utils
+import pytest
 
 
 def test_to_list_with_many_return_list():
-    many = [(1, 2, 3), "abc", 1, 1.0, "a", {"a": "b"}]
+    many = [(1, 2, 3), 1, 1.0]
     for item in many:
         result = utils.to_list(item)
         assert isinstance(result, list)
@@ -32,6 +33,11 @@ def test_to_list_with_many_return_list():
             assert result == [1.0]
 
 
+def test_to_list_throws():
+    with pytest.raises(TypeError):
+        utils.to_list("123")
+
+
 def test_try_clear_without_ipython():
     is_notebook = utils.IS_NOTEBOOK
     utils.IS_NOTEBOOK = False