From 6c825cb6a2e73dd4bc7833992d63f4b417fcecda Mon Sep 17 00:00:00 2001
From: Safoine El khabich <safoine.ext@zenml.io>
Date: Mon, 29 Jul 2024 12:03:43 +0100
Subject: [PATCH] update the test

---
 tests/unit/utils/test_package_utils.py | 57 +++++++++++++-------------
 1 file changed, 29 insertions(+), 28 deletions(-)

diff --git a/tests/unit/utils/test_package_utils.py b/tests/unit/utils/test_package_utils.py
index 3249d4096fb..c69e9d4641c 100644
--- a/tests/unit/utils/test_package_utils.py
+++ b/tests/unit/utils/test_package_utils.py
@@ -16,38 +16,39 @@
 from zenml.utils.package_utils import clean_requirements
 
 
-@pytest.mark.parametrize("input_reqs, expected_output", [
-    (
-        ["package1==1.0.0", "package2>=2.0.0", "package3<3.0.0"],
-        ["package1==1.0.0", "package2>=2.0.0", "package3<3.0.0"]
-    ),
-    (
-        ["package1==1.0.0", "package1==2.0.0", "package2>=2.0.0"],
-        ["package1==2.0.0", "package2>=2.0.0"]
-    ),
-    (
-        ["package1[extra]==1.0.0", "package2[test,dev]>=2.0.0"],
-        ["package1[extra]==1.0.0", "package2[test,dev]>=2.0.0"]
-    ),
-    (
-        ["package1", "package2==2.0.0", "package1>=1.5.0", "package3<3.0.0"],
-        ["package1>=1.5.0", "package2==2.0.0", "package3<3.0.0"]
-    ),
-    (
-        [],
-        []
-    ),
-])
+@pytest.mark.parametrize(
+    "input_reqs, expected_output",
+    [
+        (
+            ["package1==1.0.0", "package2>=2.0.0", "package3<3.0.0"],
+            ["package1==1.0.0", "package2>=2.0.0", "package3<3.0.0"],
+        ),
+        (
+            ["package1==1.0.0", "package1==2.0.0", "package2>=2.0.0"],
+            ["package1==2.0.0", "package2>=2.0.0"],
+        ),
+        (
+            ["package1[extra]==1.0.0", "package2[test,dev]>=2.0.0"],
+            ["package1[extra]==1.0.0", "package2[test,dev]>=2.0.0"],
+        ),
+        (
+            [
+                "package1",
+                "package2==2.0.0",
+                "package1>=1.5.0",
+                "package3<3.0.0",
+            ],
+            ["package1>=1.5.0", "package2==2.0.0", "package3<3.0.0"],
+        ),
+        ([], []),
+    ],
+)
 def test_clean_requirements(input_reqs, expected_output):
     """Test clean_requirements function."""
     assert clean_requirements(input_reqs) == expected_output
 
-def test_clean_requirements_type_error():
-    """Test clean_requirements function with wrong input type."""
-    with pytest.raises(TypeError):
-        clean_requirements("not a list")
 
 def test_clean_requirements_value_error():
     """Test clean_requirements function with wrong input value."""
-    with pytest.raises(ValueError):
-        clean_requirements([1, 2, 3])  # List of non-string elements
\ No newline at end of file
+    with pytest.raises(AttributeError):
+        clean_requirements([1, 2, 3])  # List of non-string elements