xrtm-data 0.2.2__tar.gz → 0.2.3__tar.gz

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xrtm-data
-Version: 0.2.2
+Version: 0.2.3
 Summary: The Snapshot Vault for XRTM.
 Author-email: XRTM Team <moy@xrtm.org>
 License-Expression: Apache-2.0

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "xrtm-data"
-version = "0.2.2"
+version = "0.2.3"
 description = "The Snapshot Vault for XRTM."
 readme = "README.md"
 requires-python = ">=3.11"

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/src/xrtm/data/cli/__init__.py

@@ -75,6 +75,9 @@ def collect(
         xrtm-data collect -m 0x1234... -d 30 -o data/trades.parquet
     """
     output_path = Path(output)
+    output_suffix = output_path.suffix.lower()
+    if output_suffix not in {".parquet", ".json"}:
+        raise click.ClickException("Output path must end with .parquet or .json")
 
     # Check cache
     if output_path.exists() and not force:
@@ -124,10 +127,12 @@ def collect(
     # Save output
     output_path.parent.mkdir(parents=True, exist_ok=True)
 
-    if output_path.suffix == ".parquet":
+    if output_suffix == ".parquet":
         _save_parquet(window, output_path)
-    else:
+    elif output_suffix == ".json":
         _save_json(window, output_path)
+    else:  # pragma: no cover - guarded before collection
+        raise click.ClickException("Output path must end with .parquet or .json")
 
     console.print(f"[green]✓ Saved {len(window.trades)} trades to:[/green] {output_path}")
 
@@ -214,8 +219,9 @@ def info(file_path: str):
     Displays summary statistics for trade files or prior files.
     """
     path = Path(file_path)
+    suffix = path.suffix.lower()
 
-    if path.suffix == ".json":
+    if suffix == ".json":
         with open(path) as f:
             data = json.load(f)
 
@@ -230,7 +236,7 @@ def info(file_path: str):
             ))
         else:
             console.print(f"JSON file with {len(data)} keys")
-    elif path.suffix == ".parquet":
+    elif suffix == ".parquet":
         import pyarrow.parquet as pq
 
         table = pq.read_table(path)
@@ -240,6 +246,8 @@ def info(file_path: str):
             f"Columns: {table.column_names}",
             title=path.name,
         ))
+    else:
+        raise click.ClickException("Unsupported file type. Expected .json or .parquet")
 
 
 def _save_parquet(window, path: Path) -> None:

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/src/xrtm/data/version.py

@@ -21,7 +21,7 @@ This module provides the single source of truth for the package version.
 
 __all__ = ["__version__", "__author__", "__contact__", "__license__", "__copyright__"]
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 __author__ = "XRTM Team"
 __contact__ = "moy@xrtm.org"
 __license__ = "Apache-2.0"

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/src/xrtm_data.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xrtm-data
-Version: 0.2.2
+Version: 0.2.3
 Summary: The Snapshot Vault for XRTM.
 Author-email: XRTM Team <moy@xrtm.org>
 License-Expression: Apache-2.0

{xrtm_data-0.2.2 → xrtm_data-0.2.3}/src/xrtm_data.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ src/xrtm_data.egg-info/requires.txt
 src/xrtm_data.egg-info/top_level.txt
 tests/test_beta_fitter.py
 tests/test_cli_loading.py
+tests/test_cli_ux.py
 tests/test_local_datasource.py
 tests/test_polymarket_source.py
 tests/test_polymarket_subgraph.py

xrtm_data-0.2.3/tests/test_cli_ux.py

@@ -0,0 +1,41 @@
+import json
+from pathlib import Path
+
+from click.testing import CliRunner
+
+from xrtm.data.cli import main
+
+
+def test_info_rejects_unsupported_suffix() -> None:
+    runner = CliRunner()
+
+    with runner.isolated_filesystem():
+        path = Path("trades.txt")
+        path.write_text("not a supported data file")
+
+        result = runner.invoke(main, ["info", str(path)])
+
+        assert result.exit_code != 0
+        assert "Unsupported file type" in result.output
+
+
+def test_collect_rejects_unsupported_output_suffix_before_fetching() -> None:
+    runner = CliRunner()
+
+    result = runner.invoke(main, ["collect", "-m", "0x1234", "-o", "trades.txt"])
+
+    assert result.exit_code != 0
+    assert "Output path must end with .parquet or .json" in result.output
+
+
+def test_info_accepts_json_file() -> None:
+    runner = CliRunner()
+
+    with runner.isolated_filesystem():
+        path = Path("prior.json")
+        path.write_text(json.dumps({"family": "beta", "alpha": 1, "beta": 2}))
+
+        result = runner.invoke(main, ["info", str(path)])
+
+        assert result.exit_code == 0
+        assert "Prior File" in result.output