add more config tests

Author: geetu040

tests/test_openml/test_config.py

@@ -190,3 +190,40 @@ def test_openml_cache_dir_env_var(tmp_path: Path) -> None:
 
         assert openml.config._root_cache_directory == expected_path
         assert openml.config.get_cache_directory() == str(expected_path / "org" / "openml" / "www")
+
+
+@pytest.mark.parametrize("mode", ["production", "test", "local"])
+@pytest.mark.parametrize("api_version", [APIVersion.V1, APIVersion.V2])
+def test_get_servers(mode, api_version):
+    orig_servers = openml.config.get_servers(mode)
+
+    openml.config.set_servers(mode)
+    openml.config.set_api_version(api_version)
+    openml.config.server = "temp-server1"
+    openml.config.apikey = "temp-apikey1"
+    openml.config.get_servers(mode)["server"] = 'temp-server2'
+    openml.config.get_servers(mode)["apikey"] = 'temp-server2'
+
+    assert openml.config.get_servers(mode) == orig_servers
+
+
+@pytest.mark.parametrize("mode", ["production", "test", "local"])
+@pytest.mark.parametrize("api_version", [APIVersion.V1, APIVersion.V2])
+def test_set_servers(mode, api_version):
+    openml.config.set_servers(mode)
+    openml.config.set_api_version(api_version)
+
+    assert openml.config.servers == openml.config.get_servers(mode)
+    assert openml.config.api_version == api_version
+
+    openml.config.server = "temp-server"
+    openml.config.apikey = "temp-apikey"
+
+    assert openml.config.server == openml.config.servers[api_version]["server"]
+    assert openml.config.apikey == openml.config.servers[api_version]["apikey"]
+
+    for version, servers in openml.config.servers.items():
+        if version == api_version:
+            assert servers != openml.config.get_servers(mode)[version]
+        else:
+            assert servers == openml.config.get_servers(mode)[version]