Skip to content

Commit ceef49d

Browse files
committed
fixed testing
1 parent 0becb6e commit ceef49d

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

tests/test_datamint_config.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)