support python -m mddatasetbuilder (#218)

Signed-off-by: Jinzhe Zeng <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
tongzhugroup · May 25, 2024 · da3a0ab · da3a0ab
1 parent 132cca8
commit da3a0ab
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/mddatasetbuilder/__main__.py b/mddatasetbuilder/__main__.py
@@ -0,0 +1,6 @@
+"""Module entry point."""
+
+from .datasetbuilder import _commandline
+
+if __name__ == "__main__":
+    _commandline()
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -0,0 +1,9 @@
+"""Test command line interface."""
+
+import subprocess as sp
+import sys
+
+
+def test_module():
+    """Test python -m mddatasetbuilder."""
+    sp.check_output([sys.executable, "-m", "mddatasetbuilder", "-h"])