mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
TPTBusiness
cbe1c52e00
Rename all source files, scripts, tests, documentation, and configuration from Predix/predix to NexQuant/nexquant across the entire codebase.
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
"""Tests for scripts/nexquant_full_eval.py pure functions and dataclasses."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
PROJECT_ROOT = Path(__file__).parent.parent.parent
|
|
sys.path.insert(0, str(PROJECT_ROOT))
|
|
|
|
|
|
class TestFactorInfo:
|
|
def test_construction(self):
|
|
from scripts.nexquant_full_eval import FactorInfo
|
|
fi = FactorInfo(
|
|
workspace_hash="abc123",
|
|
factor_name="test_factor",
|
|
factor_code="x=1",
|
|
)
|
|
assert fi.workspace_hash == "abc123"
|
|
assert fi.factor_name == "test_factor"
|
|
assert fi.factor_code == "x=1"
|
|
|
|
|
|
class TestEvalResult:
|
|
def test_defaults(self):
|
|
from scripts.nexquant_full_eval import EvalResult
|
|
er = EvalResult(factor_name="f1", workspace_hash="h1")
|
|
assert er.status == ""
|
|
assert er.ic is None
|
|
assert er.error_message is None
|
|
assert er.non_null_count == 0
|
|
|
|
def test_failed_result(self):
|
|
from scripts.nexquant_full_eval import EvalResult
|
|
er = EvalResult(
|
|
factor_name="f1", workspace_hash="h1",
|
|
status="failed", error_message="timeout",
|
|
)
|
|
assert er.status == "failed"
|
|
assert er.error_message == "timeout"
|
|
|
|
def test_to_dict(self):
|
|
from scripts.nexquant_full_eval import EvalResult
|
|
er = EvalResult(factor_name="f1", workspace_hash="h1", status="success", ic=0.05)
|
|
d = er.to_dict()
|
|
assert d["factor_name"] == "f1"
|
|
assert d["ic"] == 0.05
|
|
assert d["status"] == "success"
|
|
|
|
|
|
class TestExtractFactorDescription:
|
|
def test_docstring_extracted(self):
|
|
from scripts.nexquant_full_eval import _extract_factor_description
|
|
code = '"""This is a test factor.\nComputes momentum."""\nx=1'
|
|
desc = _extract_factor_description(code)
|
|
assert "test factor" in desc
|
|
|
|
def test_comment_extraction(self):
|
|
from scripts.nexquant_full_eval import _extract_factor_description
|
|
code = "# Momentum factor\n# Uses 20-bar window\nx=1"
|
|
desc = _extract_factor_description(code)
|
|
assert "Momentum factor" in desc
|
|
assert "20-bar window" in desc
|
|
|
|
def test_no_docstring_or_comments(self):
|
|
from scripts.nexquant_full_eval import _extract_factor_description
|
|
code = "x = 1\ny = 2\n"
|
|
desc = _extract_factor_description(code)
|
|
assert desc == "No description available"
|
|
|
|
def test_shebang_skipped(self):
|
|
from scripts.nexquant_full_eval import _extract_factor_description
|
|
code = "#!/usr/bin/env python\n# Real comment\nx=1"
|
|
desc = _extract_factor_description(code)
|
|
assert "Real comment" in desc
|
|
assert "usr/bin" not in desc
|