Add doctest for infer_compression. Increases coverage from 77% to 80%.

jaraco · Jan 19, 2023 · 80f4d01 · 80f4d01
1 parent 2706e47
commit 80f4d01
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/jaraco/context.py b/jaraco/context.py
@@ -55,6 +55,15 @@ def tarball_context(url, target_dir=None, runner=None, pushd=pushd):
 def infer_compression(url):
     """
     Given a URL or filename, infer the compression code for tar.
+
+    >>> infer_compression('http://foo/bar.tar.gz')
+    'z'
+    >>> infer_compression('http://foo/bar.tgz')
+    'z'
+    >>> infer_compression('file.bz')
+    'j'
+    >>> infer_compression('file.xz')
+    'J'
     """
     # cheat and just assume it's the last two characters
     compression_indicator = url[-2:]