Skip to content

Commit 19e83d6

Browse files
author
Tyler Coles
committed
Fix Windows test bug.
1 parent 7048895 commit 19e83d6

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

tests/fast/tools/data_test.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -48,15 +48,15 @@ def dataframe(self) -> pd.DataFrame:
4848
rume,
4949
pd.DataFrame(
5050
{
51-
"tick": np.repeat(np.arange(0, 4), 2),
51+
"tick": np.repeat(np.arange(0, 4, dtype=np.int64), 2),
5252
"date": np.repeat(np.arange(date(2021, 1, 1), date(2021, 1, 5)), 2),
5353
"node": np.tile(["04", "35"], 4),
54-
"S": np.arange(3000, 3800, 100),
55-
"I": np.arange(2000, 2800, 100),
56-
"R": np.arange(1000, 1800, 100),
57-
"S → I": np.arange(300, 380, 10),
58-
"I → R": np.arange(200, 280, 10),
59-
"R → S": np.arange(100, 180, 10),
54+
"S": np.arange(3000, 3800, 100, dtype=np.int64),
55+
"I": np.arange(2000, 2800, 100, dtype=np.int64),
56+
"R": np.arange(1000, 1800, 100, dtype=np.int64),
57+
"S → I": np.arange(300, 380, 10, dtype=np.int64),
58+
"I → R": np.arange(200, 280, 10, dtype=np.int64),
59+
"R → S": np.arange(100, 180, 10, dtype=np.int64),
6060
}
6161
),
6262
)
@@ -73,11 +73,11 @@ def test_basic_munge(rume, output):
7373

7474
expected = pd.DataFrame(
7575
{
76-
"time": [1, 2],
76+
"time": np.array([1, 2], dtype=np.int64),
7777
"geo": ["04", "04"],
78-
"S": [3200, 3400],
79-
"I": [2200, 2400],
80-
"R": [1200, 1400],
78+
"S": np.array([3200, 3400], dtype=np.int64),
79+
"I": np.array([2200, 2400], dtype=np.int64),
80+
"R": np.array([1200, 1400], dtype=np.int64),
8181
}
8282
)
8383

0 commit comments

Comments
 (0)