Unit test to validate Cluster configuration parameters

Author: Ygnas

tests/unit_test.py

@@ -67,6 +67,7 @@
 from tests.unit_test_support import (
     createClusterWithConfig,
     createClusterConfig,
+    createClusterWrongType,
 )
 
 import codeflare_sdk.utils.kube_api_helpers
@@ -267,6 +268,10 @@ def test_config_creation():
     assert config.image_pull_secrets == ["unit-test-pull-secret"]
     assert config.appwrapper == True
 
+def test_config_creation_wrong_type():
+    with pytest.raises(TypeError):
+        config = createClusterWrongType()
+
 
 def test_cluster_creation(mocker):
     # Create AppWrapper containing a Ray Cluster with no local queue specified

tests/unit_test_support.py

@@ -31,3 +31,21 @@ def createClusterWithConfig(mocker):
     )
     cluster = Cluster(createClusterConfig())
     return cluster
+
+def createClusterWrongType():
+    config = ClusterConfiguration(
+        name="unit-test-cluster",
+        namespace="ns",
+        num_workers=2,
+        worker_cpu_requests=[],
+        worker_cpu_limits=4,
+        worker_memory_requests=5,
+        worker_memory_limits=6,
+        worker_extended_resource_requests={"nvidia.com/gpu": 7},
+        appwrapper=True,
+        machine_types=["cpu.small", "gpu.large"],
+        image_pull_secrets=["unit-test-pull-secret"],
+        image="quay.io/rhoai/ray:2.23.0-py39-cu121",
+        write_to_file=True,
+    )
+    return config