From 734221e32d6f3d0cbff8a47cf6eead0f8cf067da Mon Sep 17 00:00:00 2001
From: Kou-Kinoshita <lucas.kou@aignosi.com.br>
Date: Mon, 27 Oct 2025 19:45:47 -0300
Subject: [PATCH] SIENTIAPDE-1321: Updated tests

---
 tests/utils/models/test_train_model_result.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/utils/models/test_train_model_result.py b/tests/utils/models/test_train_model_result.py
index 0ba7912..762ee64 100644
--- a/tests/utils/models/test_train_model_result.py
+++ b/tests/utils/models/test_train_model_result.py
@@ -175,11 +175,11 @@ def test_train_model_result_is_dataclass(sample_params, sample_dataframes):
 
 
 def test_train_model_result_field_count():
-    """Test that TrainModelResult has exactly 17 fields."""
+    """Test that TrainModelResult has exactly 19 fields."""
     from dataclasses import fields
 
     result_fields = fields(TrainModelResult)
-    assert len(result_fields) == 17
+    assert len(result_fields) == 19
 
     field_names = {f.name for f in result_fields}
     expected_fields = {
@@ -195,6 +195,8 @@ def test_train_model_result_field_count():
         'mse_val',
         'mae_val',
         'r2_val',
+        'equation',
+        'equation_path',
         'run_name',
         'report_path',
         'train_data_path',