Skip to content

Commit 8cac767

Browse files
committed
Added some NaNs to tests to verify behavior
1 parent e472089 commit 8cac767

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

_delphi_utils_python/tests/test_archive.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from boto3 import Session
77
from git import Repo, exc
88
from moto import mock_s3
9+
import numpy as np
910
import pandas as pd
1011
from pandas.testing import assert_frame_equal
1112
import pytest
@@ -43,7 +44,7 @@ def test_diff_and_filter_exports(self, tmp_path):
4344
"csv1": pd.DataFrame({
4445
"geo_id": ["1", "2", "3"],
4546
"val": [1.0, 2.0, 3.0],
46-
"se": [0.1, 0.2, 0.3],
47+
"se": [np.nan, 0.2, 0.3],
4748
"sample_size": [10.0, 20.0, 30.0]}),
4849

4950
# Deleted
@@ -65,7 +66,7 @@ def test_diff_and_filter_exports(self, tmp_path):
6566
"csv1": pd.DataFrame({
6667
"geo_id": ["1", "2", "4"],
6768
"val": [1.0, 2.1, 4.0],
68-
"se": [0.1, 0.21, 0.4],
69+
"se": [np.nan, 0.21, np.nan],
6970
"sample_size": [10.0, 21.0, 40.0]}),
7071

7172
# Added
@@ -78,7 +79,7 @@ def test_diff_and_filter_exports(self, tmp_path):
7879
csv1_diff = pd.DataFrame({
7980
"geo_id": ["2", "4"],
8081
"val": [2.1, 4.0],
81-
"se": [0.21, 0.4],
82+
"se": [0.21, np.nan],
8283
"sample_size": [21.0, 40.0]})
8384

8485
arch_diff = ArchiveDiffer(cache_dir, export_dir)

0 commit comments

Comments
 (0)