Skip to content

Commit

Permalink
Update test_echo_class.py
Browse files Browse the repository at this point in the history
  • Loading branch information
lauratomkins committed Nov 13, 2023
1 parent 07f4f2d commit 6898752
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions tests/retrieve/test_echo_class.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,15 +67,15 @@ def test_steiner_conv_strat_modify_area(area_relation):
assert eclass["data"].max() == 2


def test_conv_strat__yuter_default():
def test_feature_detection_default():
grid = pyart.testing.make_storm_grid()
dict = pyart.retrieve.conv_strat_yuter(grid, bkg_rad_km=50)
dict = pyart.retrieve.feature_detection(grid, bkg_rad_km=50)

assert "convsf" in dict.keys()
assert "convsf_under" in dict.keys()
assert "convsf_over" in dict.keys()
assert "feature_detection" in dict.keys()
assert "feature_under" in dict.keys()
assert "feature_over" in dict.keys()
assert np.all(
dict["convsf"]["data"][25]
dict["feature_detection"]["data"][0, 25, :]
== np.array(
[
0,
Expand Down Expand Up @@ -123,13 +123,13 @@ def test_conv_strat__yuter_default():
)


def test_conv_strat_yuter_noest():
def test_feature_detection_noest():
grid = pyart.testing.make_storm_grid()
dict = pyart.retrieve.conv_strat_yuter(grid, bkg_rad_km=50, estimate_flag=False)
dict = pyart.retrieve.feature_detection(grid, bkg_rad_km=50, estimate_flag=False)

assert "convsf" in dict.keys()
assert "convsf_under" not in dict.keys()
assert "convsf_over" not in dict.keys()
assert "feature_detection" in dict.keys()
assert "feature_under" not in dict.keys()
assert "feature_over" not in dict.keys()


def test_hydroclass_semisupervised():
Expand Down

0 comments on commit 6898752

Please sign in to comment.