A few small fixes of the integratiom tests. Postpones rest

Author: sveinugu

tests/integration/novel/cases/raw/flows.py

@@ -53,6 +53,6 @@ def specialize_record_models_func_flow(
         tables: Dataset[GeneralTable]) -> MultiModelDataset[GeneralTable]:
 
     record_models = Dataset[RecordSchema]([
-        (table_name, extract_record_model(table)) for table_name, table in tables
+        (table_name, extract_record_model(table)) for table_name, table in tables.items()
     ])
     return apply_models_to_dataset(tables, record_models)

tests/integration/novel/cases/raw/tasks.py

@@ -27,7 +27,7 @@ def merge_key_value_into_str(key: Any, val: Any) -> str:
 @TaskTemplate()
 def extract_record_model(table: GeneralTable) -> RecordSchema:
     record_model = {}
-    for record in table:
+    for record in table.to_data():
         for field_key, field_val in record.items():
             if field_key not in record_model:
                 record_model[field_key] = type(field_val)

tests/integration/novel/helpers/models.py

@@ -33,7 +33,7 @@ class GeneralRecord(Model[Dict[str, Union[int, str]]]):
     """This is a general record"""
 
 
-class GeneralTable(TableTemplate[GeneralRecord]):
+class GeneralTable(Model[TableTemplate[GeneralRecord]]):
     """This is a general table"""
 
 

tests/integration/novel/test_multi_model_dataset.py

@@ -1,3 +1,4 @@
+import inspect
 from inspect import Parameter
 from types import NoneType
 from typing import Annotated
@@ -90,15 +91,15 @@ def test_specialize_record_models_signature_and_return_type_func(
         runtime_all_engines: Annotated[NoneType, pytest.fixture],  # noqa
         case: FlowCase):
     for flow_obj in case.flow_template, case.flow_template.apply():
-        assert flow_obj.param_signatures == {
+        assert inspect.signature(flow_obj).parameters == {
             'tables':
                 Parameter(
                     'tables',
                     Parameter.POSITIONAL_OR_KEYWORD,
                     annotation=Dataset[GeneralTable],
                 )
         }
-        assert flow_obj.return_type == MultiModelDataset[GeneralTable]
+        assert inspect.signature(flow_obj).return_annotation == MultiModelDataset[GeneralTable]
 
 
 @pc.parametrize_with_cases('case', cases='.cases.flows', has_tag='specialize_record_models')