Add test cases for format module

1 files changed, 18 insertions, 0 deletions

diff --git a/tests/test_format.py b/tests/test_format.py
new file mode 100644
index 00000000..54028724
--- /dev/null
+++ b/tests/test_format.py
@@ -0,0 +1,18 @@
+from unittest.mock import MagicMock, patch
+
+import pytest
+from hypothesis_auto import auto_pytest_magic
+
+import isort.format
+
+auto_pytest_magic(isort.format.show_unified_diff)
+
+
+def test_ask_whether_to_apply_changes_to_file():
+    with patch("isort.format.input", MagicMock(return_value="y")):
+        assert isort.format.ask_whether_to_apply_changes_to_file("")
+    with patch("isort.format.input", MagicMock(return_value="n")):
+        assert not isort.format.ask_whether_to_apply_changes_to_file("")
+    with patch("isort.format.input", MagicMock(return_value="q")):
+        with pytest.raises(SystemExit):
+            assert isort.format.ask_whether_to_apply_changes_to_file("")