File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -738,10 +738,11 @@ def test_pandas_historical_input(operator_setup, model):
738
738
ForecastOperatorConfig ,
739
739
)
740
740
741
- tmpdirname = operator_setup
742
- historical_data_path , additional_data_path = setup_artificial_data ()
741
+ historical_data_path , additional_data_path , _ = setup_artificial_data (
742
+ operator_setup
743
+ )
743
744
yaml_i , output_data_path = populate_yaml (
744
- tmpdirname = tmpdirname ,
745
+ operator_setup = operator_setup ,
745
746
historical_data_path = historical_data_path ,
746
747
additional_data_path = additional_data_path ,
747
748
)
@@ -755,7 +756,7 @@ def test_pandas_historical_input(operator_setup, model):
755
756
operator_config = ForecastOperatorConfig .from_dict (yaml_i )
756
757
operate (operator_config )
757
758
assert pd .read_csv (additional_data_path )["Date" ].equals (
758
- pd .read_csv (f"{ tmpdirname } /results/forecast.csv" )["Date" ]
759
+ pd .read_csv (f"{ operator_setup } /results/forecast.csv" )["Date" ]
759
760
)
760
761
761
762
You can’t perform that action at this time.
0 commit comments