@@ -1296,7 +1296,7 @@ def test_retrieve_methods_in_target_unit(
12961296 res = retrieve_method (external_id = ts .external_id , end = 3 , ** kwargs )
12971297
12981298 if isinstance (res , pd .DataFrame ):
1299- res = DatapointsArray (value = res .values )
1299+ res = DatapointsArray (value = res .values . ravel () )
13001300
13011301 assert math .isclose (res .value [0 ], - 40 )
13021302 assert math .isclose (res .value [1 ], 32 )
@@ -2054,14 +2054,13 @@ def test_retrieve_methods_in_target_unit(
20542054 self ,
20552055 all_retrieve_endpoints : list [Callable ],
20562056 kwargs : dict ,
2057- cognite_client : CogniteClient ,
20582057 timeseries_degree_c_minus40_0_100 : TimeSeries ,
20592058 ) -> None :
20602059 ts = timeseries_degree_c_minus40_0_100
20612060 for retrieve_method in all_retrieve_endpoints :
20622061 res = retrieve_method (external_id = ts .external_id , aggregates = "max" , granularity = "1h" , end = 3 , ** kwargs )
20632062 if isinstance (res , pd .DataFrame ):
2064- res = DatapointsArray (max = res .values )
2063+ res = DatapointsArray (max = res .values . ravel () )
20652064 assert math .isclose (res .max [0 ], 212 )
20662065
20672066 def test_status_codes_affect_aggregate_calculations (
0 commit comments