gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1241: Formatted files

Browse Source
This commit is contained in:
Kou-Kinoshita
2025-10-29 11:55:03 -03:00
parent e8250d8dd7
commit 5e763a8a64
3 changed files with 32 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/utils/repository/test_training_repository.py
View File

@@ -80,7 +80,9 @@ class TestExtractModelEquation:
assert result['coefficients']['var2'] == -0.75
assert result['coefficients']['var3'] == 2.25
def test_extract_equation_string_format(self, training_repo, sample_params, sample_linear_model):
def test_extract_equation_string_format(
self, training_repo, sample_params, sample_linear_model
):
"""Test equation string is formatted correctly."""
result = training_repo._extract_model_equation(sample_linear_model, sample_params)
@@ -93,9 +95,7 @@ class TestExtractModelEquation:
"""Test LaTeX equation is formatted correctly."""
result = training_repo._extract_model_equation(sample_linear_model, sample_params)
expected_latex = (
'target = 10.500000 + 1.500000 \\cdot var1 + -0.750000 \\cdot var2 + 2.250000 \\cdot var3'
)
expected_latex = 'target = 10.500000 + 1.500000 \\cdot var1 + -0.750000 \\cdot var2 + 2.250000 \\cdot var3'
assert result['latex_equation'] == expected_latex
def test_extract_equation_single_variable(self, training_repo, sample_linear_model):
@@ -464,4 +464,3 @@ class TestAfterTrainCalculation:
# Verify inverse_transform was called
assert mock_scaler.inverse_transform.called
assert result.y_pred is not None