@@ -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