def test_pittsburg1950():
# Perform a test of the Pitsburgh1950 dataset.
dataset_dict = {
"name": "Pittsburgh, PA",
"country": "USA",
"elevation": 310.0,
"start_year": "1950",
"end_year": "1950",
"ew_hemisphere": "W",
"is_metric": True,
"latitude": 40.45,
"longitude": 80.0,
"ns_hemisphere": "N",
"precipitation": [158.24, 76.96, 89.66, 66.29, 92.46, 118.11, 127.25, 97.54, 94.23, 33.02, 187.96, 63.5],
"temperature": [5.89, 1.33, 2.56, 8.5, 17.39, 21.28, 22.56, 22.33, 18.5, 14.72, 4.83, -1.22]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds, water_holding_capacity=76.2).to_dict() # Non-standard WHC.
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Udic",
"temperature_regime": "Mesic",
"regime_subdivision_1": "Typic", # Additional checks on regime subdivisions here.
"regime_subdivision_2": "Udic",
"annual_rainfall_mm": 1205.2,
"num_cumulative_days_dry": 0,
"num_cumulative_days_moist_dry": 13,
"num_cumulative_days_moist": 347,
"num_cumulative_days_dry_over_5c": 0,
"num_cumulative_days_moist_dry_over_5c": 13,
"num_cumulative_days_moist_over_5c": 204,
"num_consecutive_days_moist_someplaces": 360,
"num_consecutive_days_moist_over_8c_someplaces": 204,
"days_dry_after_summer_solstice": 0,
"moist_days_after_winter_solstice": 120,
"mean_potential_evapotranspiration": [15.7, 2.2, 6.4, 33.6, 96.3, 126.7, 139.0,
127.4, 87.6, 59.9, 12.0, 0.0]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_piedmont1937():
# Perform a test of the Piedmont1937 dataset.
dataset_dict = {
"name": "Piedmont Research Station, NC",
"country": "USA",
"elevation": 158.0,
"start_year": "1937",
"end_year": "1937",
"ew_hemisphere": "E",
"is_metric": True,
"latitude": 38.23,
"longitude": 78.12,
"ns_hemisphere": "N",
"precipitation": [181.1, 65.79, 34.29, 205.49, 84.07, 144.02, 123.19, 156.21, 61.21, 200.91, 62.23, 14.48],
"temperature": [5.33, 1.67, 5.83, 11.5, 18.39, 23.67, 24.83, 24.89, 18.33, 12.11, 6.61, 1.94]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds, water_holding_capacity=154.9).to_dict() # Non-standard WHC.
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Udic",
"temperature_regime": "Thermic",
"regime_subdivision_1": "Typic", # Additional checks on regime subdivisions here.
"regime_subdivision_2": "Udic",
"annual_rainfall_mm": 1333.0,
"num_cumulative_days_dry": 0,
"num_cumulative_days_moist_dry": 0,
"num_cumulative_days_moist": 360,
"num_cumulative_days_dry_over_5c": 0,
"num_cumulative_days_moist_dry_over_5c": 0,
"num_cumulative_days_moist_over_5c": 246,
"num_consecutive_days_moist_someplaces": 360,
"num_consecutive_days_moist_over_8c_someplaces": 222,
"days_dry_after_summer_solstice": 0,
"moist_days_after_winter_solstice": 120,
"mean_potential_evapotranspiration": [11.7, 2.2, 16.1, 45.3, 98.8, 142.6,
153.9, 144.5, 83.1, 42.5, 15.7, 2.7]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_column97():
# Perform a test of the Column97 dataset.
dataset_dict = {
"name": "Columbus 3 NE",
"country": "USA",
"elevation": 441.96,
"start_year": "1997",
"end_year": "1997",
"ew_hemisphere": "W",
"is_metric": True,
"latitude": 41.4666,
"longitude": 97.3333,
"ns_hemisphere": "N",
"precipitation": [7.6, 19.8, 4.1, 81.8, 86.9, 97.8, 116.6, 55.1, 103.6, 79.5, 3.9, 9.9],
"temperature": [-5.0, -1.7, 3.9, 6.1, 12.8, 20.0, 23.3, 21.7, 18.3, 10.6, 1.7, -1.7]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Udic",
"temperature_regime": "Mesic",
"annual_rainfall_mm": 666.6,
"num_cumulative_days_dry": 0,
"num_cumulative_days_moist_dry": 0,
"num_cumulative_days_moist": 360,
"num_cumulative_days_dry_over_5c": 0,
"num_cumulative_days_moist_dry_over_5c": 0,
"num_cumulative_days_moist_over_5c": 203,
"num_consecutive_days_moist_someplaces": 360,
"num_consecutive_days_moist_over_8c_someplaces": 181,
"days_dry_after_summer_solstice": 0,
"moist_days_after_winter_solstice": 120,
"mean_potential_evapotranspiration": [0.0, 0.0, 14.2, 26.0, 71.0, 121.9, 147.4, 126.9,
90.5, 43.5, 4.2, 0.0]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_chadron():
# Perform a test of the Chadron dataset.
dataset_dict = {
"name": "Chadron, NE",
"country": "USA",
"elevation": 3510.0,
"start_year": "1971",
"end_year": "2000",
"ew_hemisphere": "E",
"is_metric": False,
"latitude": 42.82,
"longitude": -103.0,
"ns_hemisphere": "N",
"precipitation": [0.46, 0.47, 0.91, 1.89, 3.02, 2.62, 2.11, 1.67, 1.44, 1.05, 0.57, 0.42],
"temperature": [22.8, 28.1, 36.2, 45.9, 56.8, 67.2, 74.1, 73.0, 61.7, 48.9, 33.7, 25.1]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Aridic",
"temperature_regime": "Mesic",
"annual_rainfall_mm": 422.4,
"num_cumulative_days_dry": 153,
"num_cumulative_days_moist_dry": 124,
"num_cumulative_days_moist": 83,
"num_cumulative_days_dry_over_5c": 103,
"num_cumulative_days_moist_dry_over_5c": 19,
"num_cumulative_days_moist_over_5c": 79,
"num_consecutive_days_moist_someplaces": 207,
"num_consecutive_days_moist_over_8c_someplaces": 87,
"days_dry_after_summer_solstice": 93,
"moist_days_after_winter_solstice": 15,
"mean_potential_evapotranspiration": [0.0, 0.0, 7.8, 35.1, 78.5, 119.8, 149.4,
134.6, 80.2, 37.5, 2.1, 0.0]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_mead89():
# Perform a test of the MEAD89 dataset.
dataset_dict = {
"name": "Mead Agronomy Lab",
"country": "USA",
"elevation": 1180.0,
"start_year": "1989",
"end_year": "1989",
"ew_hemisphere": "W",
"is_metric": False,
"latitude": 41.166666666666664,
"longitude": 96.41666666666667,
"ns_hemisphere": "N",
"precipitation": [0.8, 0.46, 0.06, 0.99, 4.08, 4.14, 3.26, 1.62, 2.94, 0.85, 0.05, 0.66],
"temperature": [32.0, 15.0, 37.0, 55.0, 63.0, 69.0, 76.0, 73.0, 63.0, 54.0, 36.0, 17.0]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Ustic",
"temperature_regime": "Mesic",
"annual_rainfall_mm": 505.7,
"num_cumulative_days_dry": 49,
"num_cumulative_days_moist_dry": 311,
"num_cumulative_days_moist": 0,
"num_cumulative_days_dry_over_5c": 49,
"num_cumulative_days_moist_dry_over_5c": 166,
"num_cumulative_days_moist_over_5c": 0,
"num_consecutive_days_moist_someplaces": 297,
"num_consecutive_days_moist_over_8c_someplaces": 88,
"days_dry_after_summer_solstice": 46,
"moist_days_after_winter_solstice": 0,
"mean_potential_evapotranspiration": [0.0, 0.0, 7.4, 58.2, 96.6, 122.5,
154.7, 132.2, 80.4, 47.5, 4.4, 0.0]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_pendleton():
# Perform a test of the Pendleton dataset.
dataset_dict = {
"name": "Pendleton, OR",
"country": "USA",
"elevation": 75.0,
"start_year": "1971",
"end_year": "2000",
"ew_hemisphere": "E",
"is_metric": False,
"latitude": 33.3,
"longitude": -117.35,
"ns_hemisphere": "N",
"precipitation": [3.00, 2.91, 2.67, 0.81, 0.32, 0.14, 0.08, 0.02, 0.14, 0.46, 0.93, 1.73],
"temperature": [55.1, 56.0, 56.8, 59.6, 63.3, 66.7, 70.5, 71.9, 70.6, 65.5, 59.1, 55.2]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Xeric",
"temperature_regime": "Thermic",
"annual_rainfall_mm": 335.5,
"num_cumulative_days_dry": 199,
"num_cumulative_days_moist_dry": 51,
"num_cumulative_days_moist": 110,
"num_cumulative_days_dry_over_5c": 199,
"num_cumulative_days_moist_dry_over_5c": 51,
"num_cumulative_days_moist_over_5c": 110,
"num_consecutive_days_moist_someplaces": 161,
"num_consecutive_days_moist_over_8c_someplaces": 161,
"days_dry_after_summer_solstice": 120,
"moist_days_after_winter_solstice": 75,
"mean_potential_evapotranspiration": [33.8, 35.3, 44.7, 56.9, 77.2, 93.1,
113.3, 113.6, 96.1, 70.8, 44.5, 33.2]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_ajo20008():
# Perform a test of the Ajo20008 dataset.
dataset_dict = {
"name": "Ajo, AZ",
"country": "USA",
"elevation": 549.0,
"start_year": "1971",
"end_year": "2000",
"ew_hemisphere": "E",
"is_metric": True,
"latitude": 32.37,
"longitude": -112.87,
"ns_hemisphere": "N",
"precipitation": [17.02, 17.02, 19.3, 5.84, 3.81, 1.27, 19.3, 41.15, 20.83, 16.0, 12.7, 22.35],
"temperature": [12.5, 14.61, 16.78, 20.56, 24.78, 30.06, 32.17, 31.44, 29.33, 23.89, 17.22, 12.61]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Aridic",
"temperature_regime": "Hyperthermic",
"annual_rainfall_mm": 196.6,
"num_cumulative_days_dry": 360,
"num_cumulative_days_moist_dry": 0,
"num_cumulative_days_moist": 0,
"num_cumulative_days_dry_over_5c": 360,
"num_cumulative_days_moist_dry_over_5c": 0,
"num_cumulative_days_moist_over_5c": 0,
"num_consecutive_days_moist_someplaces": 0,
"num_consecutive_days_moist_over_8c_someplaces": 0,
"days_dry_after_summer_solstice": 120,
"moist_days_after_winter_solstice": 0,
"mean_potential_evapotranspiration": [16.3, 23.8, 41.3, 74.4, 135.0, 192.9,
209.5, 193.2, 160.1, 100.9, 37.8, 16.3]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)
def test_column98():
# Perform a test of the Column98 dataset.
dataset_dict = {
"name": "Columbus 3 NE",
"country": "USA",
"elevation": 441.96,
"start_year": "1998",
"end_year": "1998",
"ew_hemisphere": "W",
"is_metric": True,
"latitude": 41.4666,
"longitude": 97.3333,
"ns_hemisphere": "N",
"precipitation": [17.0, 11.2, 73.4, 90.7, 64.5, 183.6, 86.6, 84.3, 28.4, 68.1, 28.4, 6.4],
"temperature": [-2.8, 1.7, -0.6, 8.3, 16.1, 17.8, 24.4, 23.3, 21.1, 10.6, 5.0, -1.1]
}
test_ds = Dataset(dataset_dict)
test_results = run_simulation(test_ds).to_dict()
# Enforce that results match against this profile.
expected_results = {
"moisture_regime": "Udic",
"temperature_regime": "Mesic",
"annual_rainfall_mm": 742.6,
"num_cumulative_days_dry": 0,
"num_cumulative_days_moist_dry": 24,
"num_cumulative_days_moist": 336,
"num_cumulative_days_dry_over_5c": 0,
"num_cumulative_days_moist_dry_over_5c": 24,
"num_cumulative_days_moist_over_5c": 188,
"num_consecutive_days_moist_someplaces": 360,
"num_consecutive_days_moist_over_8c_someplaces": 196,
"days_dry_after_summer_solstice": 0,
"moist_days_after_winter_solstice": 120,
"mean_potential_evapotranspiration": [0.0, 3.4, 0.0, 34.4, 89.8, 102.8, 154.7,
136.7, 105.3, 40.6, 13.3, 0.0]
}
# Report true/false on comparison of results and expectations.
assert compare_results(test_results, expected_results)