fix: use right type in comparison

aitorres · Oct 7, 2023 · 723fb44 · 723fb44
1 parent b38c60e
commit 723fb44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -193,7 +193,7 @@ def _download_book(book_id: int) -> Optional[BytesIO]:
     monkeypatch.setattr(cli, "setup_settings", lambda: None)
     monkeypatch.setattr(cli, "download_book", _download_book)
     monkeypatch.setattr("getpass.getpass", _getpass_mock)
-    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != "5678")
+    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != 5678)
 
     with patch.object(
         sys, "argv", ["gutenberg2kindle", "send", "--book-id", "1234", "5678", "9101"]
@@ -225,7 +225,7 @@ def _download_book(book_id: int) -> Optional[BytesIO]:
     monkeypatch.setattr(cli, "setup_settings", lambda: None)
     monkeypatch.setattr(cli, "download_book", _download_book)
     monkeypatch.setattr("getpass.getpass", _getpass_mock)
-    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != "5678")
+    monkeypatch.setattr(cli, "send_book", lambda book_id, *_: book_id != 5678)
 
     with patch.object(
         sys,