Make TOMLDecodeError show 'tomli' as its module (#135)

* Make TOMLDecodeError show 'tomli' as its module * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Move __module__ assignment to __init__, add test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
hukkin · Oct 22, 2021 · 96dfe2c · 96dfe2c
1 parent 8917fe3
commit 96dfe2c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/tests/test_error.py b/tests/test_error.py
@@ -31,3 +31,7 @@ def test_invalid_char_quotes():
     with pytest.raises(tomli.TOMLDecodeError) as exc_info:
         tomli.loads("v = '\n'")
     assert " '\\n' " in str(exc_info.value)
+
+
+def test_module_name():
+    assert tomli.TOMLDecodeError().__module__ == "tomli"
diff --git a/tomli/__init__.py b/tomli/__init__.py
@@ -4,3 +4,6 @@
 __version__ = "1.2.1"  # DO NOT EDIT THIS LINE MANUALLY. LET bump2version UTILITY DO IT
 
 from tomli._parser import TOMLDecodeError, load, loads
+
+# Pretend this exception was created here.
+TOMLDecodeError.__module__ = "tomli"