19
19
20
20
def test_grdtrack_input_dataframe_and_dataarray ():
21
21
"""
22
- Run grdtrack by passing in a pandas.DataFrame and xarray.DataArray as inputs
22
+ Run grdtrack by passing in a pandas.DataFrame and xarray.DataArray as
23
+ inputs
23
24
"""
24
25
dataframe = load_ocean_ridge_points ()
25
26
dataarray = load_earth_relief ().sel (lat = slice (- 49 , - 42 ), lon = slice (- 118 , - 107 ))
26
27
27
28
output = grdtrack (points = dataframe , grid = dataarray , newcolname = "bathymetry" )
28
29
assert isinstance (output , pd .DataFrame )
29
30
assert output .columns .to_list () == ["longitude" , "latitude" , "bathymetry" ]
30
- npt .assert_allclose (output .iloc [0 ], [- 110.9536 , - 42.2489 , - 2799.085897 ])
31
+ npt .assert_allclose (output .iloc [0 ], [- 110.9536 , - 42.2489 , - 2797.482959 ])
31
32
32
33
return output
33
34
@@ -45,7 +46,7 @@ def test_grdtrack_input_csvfile_and_dataarray():
45
46
assert os .path .exists (path = TEMP_TRACK ) # check that outfile exists at path
46
47
47
48
track = pd .read_csv (TEMP_TRACK , sep = "\t " , header = None , comment = ">" )
48
- npt .assert_allclose (track .iloc [0 ], [- 110.9536 , - 42.2489 , - 2799.085897 ])
49
+ npt .assert_allclose (track .iloc [0 ], [- 110.9536 , - 42.2489 , - 2797.482959 ])
49
50
finally :
50
51
os .remove (path = TEMP_TRACK )
51
52
@@ -62,7 +63,7 @@ def test_grdtrack_input_dataframe_and_ncfile():
62
63
output = grdtrack (points = dataframe , grid = ncfile , newcolname = "bathymetry" )
63
64
assert isinstance (output , pd .DataFrame )
64
65
assert output .columns .to_list () == ["longitude" , "latitude" , "bathymetry" ]
65
- npt .assert_allclose (output .iloc [0 ], [- 32.2971 , 37.4118 , - 1685.745884 ])
66
+ npt .assert_allclose (output .iloc [0 ], [- 32.2971 , 37.4118 , - 1686.692142 ])
66
67
67
68
return output
68
69
@@ -80,7 +81,7 @@ def test_grdtrack_input_csvfile_and_ncfile():
80
81
assert os .path .exists (path = TEMP_TRACK ) # check that outfile exists at path
81
82
82
83
track = pd .read_csv (TEMP_TRACK , sep = "\t " , header = None , comment = ">" )
83
- npt .assert_allclose (track .iloc [0 ], [- 32.2971 , 37.4118 , - 1685.745884 ])
84
+ npt .assert_allclose (track .iloc [0 ], [- 32.2971 , 37.4118 , - 1686.692142 ])
84
85
finally :
85
86
os .remove (path = TEMP_TRACK )
86
87
@@ -89,7 +90,8 @@ def test_grdtrack_input_csvfile_and_ncfile():
89
90
90
91
def test_grdtrack_wrong_kind_of_points_input ():
91
92
"""
92
- Run grdtrack using points input that is not a pandas.DataFrame (matrix) or file
93
+ Run grdtrack using points input that is not a pandas.DataFrame (matrix) or
94
+ file
93
95
"""
94
96
dataframe = load_ocean_ridge_points ()
95
97
invalid_points = dataframe .longitude .to_xarray ()
@@ -102,7 +104,8 @@ def test_grdtrack_wrong_kind_of_points_input():
102
104
103
105
def test_grdtrack_wrong_kind_of_grid_input ():
104
106
"""
105
- Run grdtrack using grid input that is not as xarray.DataArray (grid) or file
107
+ Run grdtrack using grid input that is not as xarray.DataArray (grid) or
108
+ file
106
109
"""
107
110
dataframe = load_ocean_ridge_points ()
108
111
dataarray = load_earth_relief ().sel (lat = slice (- 49 , - 42 ), lon = slice (- 118 , - 107 ))
0 commit comments