@@ -42,7 +42,7 @@ def mock_install_package():
4242def mock_distribution ():
4343 with patch .object (main , "distribution" ) as mock_distribution :
4444 mock_distribution .return_value .version = "1.20.0"
45- mock_distribution .return_value .name = "graph_sitter "
45+ mock_distribution .return_value .name = "graph-sitter "
4646 yield mock_distribution
4747
4848
@@ -64,7 +64,7 @@ def test_update_default(mock_install_package):
6464 result = runner .invoke (main .update_command , [])
6565 assert result .exit_code == 0
6666
67- mock_install_package .assert_called_once_with ("graph_sitter " , "--upgrade" )
67+ mock_install_package .assert_called_once_with ("graph-sitter " , "--upgrade" )
6868
6969
7070def test_update_with_version_flag (mock_install_package ):
@@ -77,7 +77,7 @@ def test_update_with_version_flag(mock_install_package):
7777 assert result .exit_code == 0
7878
7979 # Check that install_package was called with "graph_sitter==0.3.0"
80- mock_install_package .assert_called_once_with ("graph_sitter ==0.3.0" )
80+ mock_install_package .assert_called_once_with ("graph-sitter ==0.3.0" )
8181
8282
8383def test_list_versions (mock_request ):
@@ -90,7 +90,7 @@ def test_list_versions(mock_request):
9090 result = runner .invoke (main .update_command , ["--list" ])
9191 assert result .exit_code == 0
9292
93- mock_request .assert_called_once_with ("https://pypi.org/pypi/graph_sitter /json" )
93+ mock_request .assert_called_once_with ("https://pypi.org/pypi/graph-sitter /json" )
9494
9595 output = result .output
9696 expected_lines = [
0 commit comments