1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
# Copyright (c) 2020 MetPy Developers.
# Distributed under the terms of the BSD 3-Clause License.
# SPDX-License-Identifier: BSD-3-Clause
"""Test station data information."""
import numpy as np
from numpy.testing import assert_almost_equal
import pandas as pd
import pytest
from metpy.io import add_station_lat_lon, station_info
def test_add_lat_lon_station_data():
"""Test for when the METAR does not correspond to a station in the dictionary."""
df = pd.DataFrame({'station': ['KOUN', 'KVPZ', 'KDEN', 'PAAA']})
df = add_station_lat_lon(df, 'station')
assert_almost_equal(df.loc[df.station == 'KOUN'].latitude.values[0], 35.25)
assert_almost_equal(df.loc[df.station == 'KOUN'].longitude.values[0], -97.47)
assert_almost_equal(df.loc[df.station == 'KVPZ'].latitude.values[0], 41.45)
assert_almost_equal(df.loc[df.station == 'KVPZ'].longitude.values[0], -87)
assert_almost_equal(df.loc[df.station == 'KDEN'].latitude.values[0], 39.85)
assert_almost_equal(df.loc[df.station == 'KDEN'].longitude.values[0], -104.65)
assert_almost_equal(df.loc[df.station == 'PAAA'].latitude.values[0], np.nan)
assert_almost_equal(df.loc[df.station == 'PAAA'].longitude.values[0], np.nan)
assert df['longitude'].dtype == np.float64
def test_add_lat_lon_station_data_optional():
"""Test for when only one argument is passed."""
df = pd.DataFrame({'station': ['KOUN', 'KVPZ', 'KDEN', 'PAAA']})
df = add_station_lat_lon(df)
assert_almost_equal(df.loc[df.station == 'KOUN'].latitude.values[0], 35.25)
def test_add_lat_lon_station_data_not_found():
"""Test case where key cannot be inferred."""
df = pd.DataFrame({'location': ['KOUN']})
with pytest.raises(KeyError):
add_station_lat_lon(df)
def test_add_lat_lon_station_data_existing_col():
"""Test will fail if one of these columns exists already."""
df = pd.DataFrame({'station': ['KOUN', 'KVPZ'],
'latitude': [44, 42]})
with pytest.raises(ValueError, match='cannot insert'):
add_station_lat_lon(df)
def test_station_lookup_get_station():
"""Test that you can get a station by ID from the lookup."""
assert station_info['KOUN'].id == 'KOUN'
def test_station_lookup_len():
"""Test that you can get the length of the station data."""
assert len(station_info) == 13798
def test_station_lookup_iter():
"""Test iterating over the station data."""
for stid in station_info:
assert stid in station_info
|
