Merge pull request #545 from GavinHuttley/develop

DEV: infer doc file type from suffix if single file provided
cogent3 · Feb 23, 2020 · 078f7a8 · 078f7a8
2 parents 3ca9c14 + 035b22b
commit 078f7a8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/doc/doctest_rsts.py b/doc/doctest_rsts.py
@@ -90,6 +90,9 @@ def main(file_paths, just, exclude, exit_on_first, suffix, verbose):
     if "," in file_paths:
         file_paths = file_paths.split(",")
     elif pathlib.Path(file_paths).is_file():
+        p = pathlib.Path(file_paths)
+        assert p.suffix.lower() in ".rst.ipynb", f"Unknown format suffix '{p.suffix}'"
+        suffix = p.suffix[1:]
         file_paths = [file_paths]
     else:
         file_paths = list(pathlib.Path(file_paths).glob(f"*.{suffix}"))