6
6
from boto3 import Session
7
7
from git import Repo , exc
8
8
from moto import mock_s3
9
+ import numpy as np
9
10
import pandas as pd
10
11
from pandas .testing import assert_frame_equal
11
12
import pytest
@@ -43,7 +44,7 @@ def test_diff_and_filter_exports(self, tmp_path):
43
44
"csv1" : pd .DataFrame ({
44
45
"geo_id" : ["1" , "2" , "3" ],
45
46
"val" : [1.0 , 2.0 , 3.0 ],
46
- "se" : [0.1 , 0.2 , 0.3 ],
47
+ "se" : [np . nan , 0.2 , 0.3 ],
47
48
"sample_size" : [10.0 , 20.0 , 30.0 ]}),
48
49
49
50
# Deleted
@@ -65,7 +66,7 @@ def test_diff_and_filter_exports(self, tmp_path):
65
66
"csv1" : pd .DataFrame ({
66
67
"geo_id" : ["1" , "2" , "4" ],
67
68
"val" : [1.0 , 2.1 , 4.0 ],
68
- "se" : [0.1 , 0.21 , 0.4 ],
69
+ "se" : [np . nan , 0.21 , np . nan ],
69
70
"sample_size" : [10.0 , 21.0 , 40.0 ]}),
70
71
71
72
# Added
@@ -78,7 +79,7 @@ def test_diff_and_filter_exports(self, tmp_path):
78
79
csv1_diff = pd .DataFrame ({
79
80
"geo_id" : ["2" , "4" ],
80
81
"val" : [2.1 , 4.0 ],
81
- "se" : [0.21 , 0.4 ],
82
+ "se" : [0.21 , np . nan ],
82
83
"sample_size" : [21.0 , 40.0 ]})
83
84
84
85
arch_diff = ArchiveDiffer (cache_dir , export_dir )
0 commit comments