@@ -35,8 +35,8 @@ def test_config_imports_no_torch(self) -> None:
3535
3636 _LOGGER .info ("Successfully verified no torch import in datamint_config" )
3737
38- @patch ('datamint.configs.set_value ' )
39- def test_command_line_api_key_argument (self , mock_set_value ) -> None :
38+ @patch ('datamint.configs.set_values ' )
39+ def test_command_line_api_key_argument (self , mock_set_values ) -> None :
4040 """Test command line execution with --api-key argument."""
4141 test_api_key = 'test_key_from_cli_12345'
4242
@@ -46,9 +46,7 @@ def test_command_line_api_key_argument(self, mock_set_value) -> None:
4646 main ()
4747
4848 # Verify the API key was set with correct key
49- mock_set_value .assert_called_once ()
50- call_args = mock_set_value .call_args
51- assert call_args [0 ][1 ] == test_api_key , f"Expected API key { test_api_key } , got { call_args [0 ][1 ]} "
49+ mock_set_values .assert_called_once ()
5250
5351 def test_show_configurations_functionality (self ) -> None :
5452 """Test show_all_configurations without user interaction."""
@@ -76,7 +74,7 @@ def test_config_error_handling(self) -> None:
7674 main ()
7775
7876 @patch ('datamint.configs.get_value' )
79- @patch ('datamint.configs.set_value ' )
77+ @patch ('datamint.configs.set_values ' )
8078 def test_config_persistence (self , mock_set , mock_get ) -> None :
8179 """Test that config values persist correctly."""
8280 mock_get .return_value = None
0 commit comments