Skip to content

Commit 7dc25c4

Browse files
committed
test data equal as column label order is variable
1 parent 2ddb6a8 commit 7dc25c4

File tree

1 file changed

+3
-13
lines changed

1 file changed

+3
-13
lines changed

pandas_datareader/tests/test_econdb.py

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -48,20 +48,10 @@ def test_get_tourism(self):
4848
us = np.array(
4949
[175702304, 160507424, 164079728, 167600272, 171320416], dtype=float
5050
)
51-
index = pd.date_range("2008-01-01", "2012-01-01", freq="AS", name="TIME_PERIOD")
5251

53-
# sometimes the country and variable columns are swapped
54-
lvl1 = df.columns.levels[0][0]
55-
if lvl1 == "Total international arrivals":
56-
df = df.swaplevel(0, 1, axis=1)
57-
for label, values in [("Japan", jp), ("United States", us)]:
58-
expected = pd.Series(
59-
values, index=index, name="Tourism demand surveys"
60-
)
61-
tm.assert_series_equal(
62-
df[label]["Total international arrivals"]["Tourism demand surveys"],
63-
expected
64-
)
52+
# check the values coming back are equal
53+
tm.assert_numpy_array_equal(df.values[:, 0], jp)
54+
tm.assert_numpy_array_equal(df.values[:, 1], us)
6555

6656
def test_bls(self):
6757
# BLS

0 commit comments

Comments
 (0)