Skip to content

Adjust data checks in datasets sample tests #2223

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Dec 6, 2022
Merged

Adjust data checks in datasets sample tests #2223

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d0f0f45
Update test_datasets_samples.py
michaelgrund Dec 5, 2022
534daf0
Apply suggestions from code review
michaelgrund Dec 5, 2022
2887f7f
Merge branch 'main' into adjust-dataset-tests
michaelgrund Dec 5, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
105 changes: 48 additions & 57 deletions pygmt/tests/test_datasets_samples.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,13 +33,12 @@ def test_japan_quakes():
data = load_japan_quakes()
assert len(record) == 1
assert data.shape == (115, 7)
summary = data.describe()
assert summary.loc["min", "year"] == 1987
assert summary.loc["max", "year"] == 1988
assert summary.loc["min", "month"] == 1
assert summary.loc["max", "month"] == 12
assert summary.loc["min", "day"] == 1
assert summary.loc["max", "day"] == 31
assert data["year"].min() == 1987
assert data["year"].max() == 1988
assert data["month"].min() == 1
assert data["month"].max() == 12
assert data["day"].min() == 1
assert data["day"].max() == 31


def test_load_sample_data():
Expand All @@ -48,13 +47,12 @@ def test_load_sample_data():
"""
data = load_sample_data(name="japan_quakes")
assert data.shape == (115, 7)
summary = data.describe()
assert summary.loc["min", "year"] == 1987
assert summary.loc["max", "year"] == 1988
assert summary.loc["min", "month"] == 1
assert summary.loc["max", "month"] == 12
assert summary.loc["min", "day"] == 1
assert summary.loc["max", "day"] == 31
assert data["year"].min() == 1987
assert data["year"].max() == 1988
assert data["month"].min() == 1
assert data["month"].max() == 12
assert data["day"].min() == 1
assert data["day"].max() == 31


def test_ocean_ridge_points():
Expand All @@ -65,11 +63,10 @@ def test_ocean_ridge_points():
data = load_ocean_ridge_points()
assert len(record) == 1
assert data.shape == (4146, 2)
summary = data.describe()
assert summary.loc["min", "longitude"] == -179.9401
assert summary.loc["max", "longitude"] == 179.935
assert summary.loc["min", "latitude"] == -65.6182
assert summary.loc["max", "latitude"] == 86.8
assert data["longitude"].min() == -179.9401
assert data["longitude"].max() == 179.935
assert data["latitude"].min() == -65.6182
assert data["latitude"].max() == 86.8


def test_sample_bathymetry():
Expand All @@ -80,13 +77,12 @@ def test_sample_bathymetry():
data = load_sample_bathymetry()
assert len(record) == 1
assert data.shape == (82970, 3)
summary = data.describe()
assert summary.loc["min", "longitude"] == 245.0
assert summary.loc["max", "longitude"] == 254.705
assert summary.loc["min", "latitude"] == 20.0
assert summary.loc["max", "latitude"] == 29.99131
assert summary.loc["min", "bathymetry"] == -7708.0
assert summary.loc["max", "bathymetry"] == -9.0
assert data["longitude"].min() == 245.0
assert data["longitude"].max() == 254.705
assert data["latitude"].min() == 20.0
assert data["latitude"].max() == 29.99131
assert data["bathymetry"].min() == -7708.0
assert data["bathymetry"].max() == -9.0


def test_usgs_quakes():
Expand All @@ -107,11 +103,10 @@ def test_fractures_compilation():
data = load_fractures_compilation()
assert len(record) == 1
assert data.shape == (361, 2)
summary = data.describe()
assert summary.loc["min", "length"] == 98.6561
assert summary.loc["max", "length"] == 984.652
assert summary.loc["min", "azimuth"] == 0.0
assert summary.loc["max", "azimuth"] == 360.0
assert data["length"].min() == 98.6561
assert data["length"].max() == 984.652
assert data["azimuth"].min() == 0.0
assert data["azimuth"].max() == 360.0


def test_mars_shape():
Expand All @@ -122,13 +117,12 @@ def test_mars_shape():
data = load_mars_shape()
assert len(record) == 1
assert data.shape == (370, 3)
summary = data.describe()
assert summary.loc["min", "longitude"] == 0.008
assert summary.loc["max", "longitude"] == 359.983
assert summary.loc["min", "latitude"] == -79.715
assert summary.loc["max", "latitude"] == 85.887
assert summary.loc["min", "radius(m)"] == -6930
assert summary.loc["max", "radius(m)"] == 15001
assert data["longitude"].min() == 0.008
assert data["longitude"].max() == 359.983
assert data["latitude"].min() == -79.715
assert data["latitude"].max() == 85.887
assert data["radius(m)"].min() == -6930
assert data["radius(m)"].max() == 15001


def test_hotspots():
Expand All @@ -154,13 +148,12 @@ def test_load_notre_dame_topography():
"""
data = load_sample_data(name="notre_dame_topography")
assert data.shape == (52, 3)
summary = data.describe()
assert summary.loc["min", "x"] == 0.2
assert summary.loc["max", "x"] == 6.3
assert summary.loc["min", "y"] == 0
assert summary.loc["max", "y"] == 6.2
assert summary.loc["min", "z"] == 690
assert summary.loc["max", "z"] == 960
assert data["x"].min() == 0.2
assert data["x"].max() == 6.3
assert data["y"].min() == 0
assert data["y"].max() == 6.2
assert data["z"].min() == 690
assert data["z"].max() == 960


def test_earth_relief_holes():
Expand All @@ -181,11 +174,10 @@ def test_maunaloa_co2():
"""
data = load_sample_data(name="maunaloa_co2")
assert data.shape == (730, 2)
summary = data.describe()
assert summary.loc["min", "date"] == 1958.2027
assert summary.loc["max", "date"] == 2019.3699
assert summary.loc["min", "co2_ppm"] == 313.2
assert summary.loc["max", "co2_ppm"] == 414.83
assert data["date"].min() == 1958.2027
assert data["date"].max() == 2019.3699
assert data["co2_ppm"].min() == 313.2
assert data["co2_ppm"].max() == 414.83


def test_rock_sample_compositions():
Expand All @@ -194,10 +186,9 @@ def test_rock_sample_compositions():
"""
data = load_sample_data(name="rock_compositions")
assert data.shape == (1000, 3)
summary = data.describe()
assert summary.loc["min", "water"] == 0
assert summary.loc["max", "water"] == 1
assert summary.loc["min", "air"] == 0
assert summary.loc["max", "air"] == 0.921
assert summary.loc["min", "limestone"] == 0
assert summary.loc["max", "limestone"] == 0.981
assert data["water"].min() == 0
assert data["water"].max() == 1
assert data["air"].min() == 0
assert data["air"].max() == 0.921
assert data["limestone"].min() == 0
assert data["limestone"].max() == 0.981