Files
mouse_control/.gitignore
Huang Qi c8fa5db7d3 Add pytest suite with 40 unit and integration tests
Coverage:
- test_model: SimpleNet forward (parametrized over batch sizes and both
  unsqueezed and flat input shapes), layer dimensions, differentiability,
  and ONNX round-trip
- test_inference: load_model resolution order (bundled, cwd override,
  explicit path, missing path), and predict shape/dtype/determinism plus
  endpoint sanity across 8 cardinal/diagonal targets
- test_train: _load_csv parsing, TrajectoryDataset indexing, full train()
  pipeline producing a single-file ONNX, plus a smoke test against the
  real data shipped under data/
- test_cli: --help for the three console scripts and a real run of
  mouse-visualize via both the entry point and python -m

Wire up pytest via dependency-groups and tool.pytest.ini_options.
2026-05-12 00:33:15 +08:00

23 lines
282 B
Plaintext

# Byte-compiled / build artifacts
__pycache__/
*.py[cod]
*.egg-info/
build/
dist/
# Virtual envs / tooling
.venv/
.uv/
# Generated outputs (only at repo root; bundled assets must stay)
/trajectories.png
/mouse.onnx
/mouse.onnx.data
# IDE
.idea/
.vscode/
# pytest
.pytest_cache/