Fix & update unit tests

rapidsai · Nov 21, 2024 · f8a9351 · f8a9351
1 parent 4872e6d
commit f8a9351
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/tests/test_json.py b/tests/test_json.py
@@ -43,7 +43,9 @@ def set_cwd(cwd: os.PathLike) -> Generator:
 @pytest.mark.parametrize(
     ["unencoded", "encoded"],
     [
-        (RAPIDSPackage(), {"publishes_prereleases": True, "has_cuda_suffix": True}),
+        (RAPIDSPackage(), {"publishes_prereleases": True, "has_cuda_suffix": True, "has_conda_package": True, "has_wheel_package": True}),
+        (RAPIDSPackage(has_conda_package=False), {"publishes_prereleases": True, "has_cuda_suffix": True, "has_conda_package": False, "has_wheel_package": True}),
+        (RAPIDSPackage(has_wheel_package=False), {"publishes_prereleases": True, "has_cuda_suffix": True, "has_conda_package": True, "has_wheel_package": False}),
         (
             RAPIDSRepository(
                 packages={
@@ -58,10 +60,14 @@ def set_cwd(cwd: os.PathLike) -> Generator:
                     "package1": {
                         "publishes_prereleases": True,
                         "has_cuda_suffix": True,
+                        "has_conda_package": True,
+                        "has_wheel_package": True
                     },
                     "package2": {
                         "publishes_prereleases": False,
                         "has_cuda_suffix": False,
+                        "has_conda_package": True,
+                        "has_wheel_package": True
                     },
                 },
             },
@@ -92,6 +98,8 @@ def set_cwd(cwd: os.PathLike) -> Generator:
                             "package": {
                                 "publishes_prereleases": True,
                                 "has_cuda_suffix": True,
+                                "has_conda_package": True,
+                                "has_wheel_package": True
                             },
                         },
                     },
@@ -100,6 +108,8 @@ def set_cwd(cwd: os.PathLike) -> Generator:
                             "proprietary-package": {
                                 "publishes_prereleases": True,
                                 "has_cuda_suffix": True,
+                                "has_conda_package": True,
+                                "has_wheel_package": True
                             },
                         },
                     },
@@ -151,7 +161,9 @@ def test_metadata_encoder(unencoded, encoded):
             '"repo1":{'
             '"packages":{'
             '"package":{'
+            '"has_conda_package":true,'
             '"has_cuda_suffix":true,'
+            '"has_wheel_package":true,'
             '"publishes_prereleases":true'
             "}"
             "}"
@@ -171,7 +183,9 @@ def test_metadata_encoder(unencoded, encoded):
             '"repo2":{'
             '"packages":{'
             '"package":{'
+            '"has_conda_package":true,'
             '"has_cuda_suffix":true,'
+            '"has_wheel_package":true,'
             '"publishes_prereleases":true'
             "}"
             "}"
@@ -191,7 +205,9 @@ def test_metadata_encoder(unencoded, encoded):
             '"repo2":{'
             '"packages":{'
             '"package":{'
+            '"has_conda_package":true,'
             '"has_cuda_suffix":true,'
+            '"has_wheel_package":true,'
             '"publishes_prereleases":true'
             "}"
             "}"
@@ -211,7 +227,9 @@ def test_metadata_encoder(unencoded, encoded):
             '"repo1":{'
             '"packages":{'
             '"package":{'
+            '"has_conda_package":true,'
             '"has_cuda_suffix":true,'
+            '"has_wheel_package":true,'
             '"publishes_prereleases":true'
             "}"
             "}"
@@ -223,7 +241,9 @@ def test_metadata_encoder(unencoded, encoded):
             '"repo2":{'
             '"packages":{'
             '"package":{'
+            '"has_conda_package":true,'
             '"has_cuda_suffix":true,'
+            '"has_wheel_package":true,'
             '"publishes_prereleases":true'
             "}"
             "}"
@@ -245,7 +265,9 @@ def test_metadata_encoder(unencoded, encoded):
                         "repo1": {
                           "packages": {
                             "package": {
+                              "has_conda_package": true,
                               "has_cuda_suffix": true,
+                              "has_wheel_package": true,
                               "publishes_prereleases": true
                             }
                           }