4
4
from pandas import (DataFrame , date_range , Timestamp , Series ,
5
5
to_datetime )
6
6
7
- from pandas .util .testing import assert_frame_equal , assert_series_equal
8
7
import pandas .util .testing as tm
9
8
10
9
from .common import TestData
@@ -51,19 +50,19 @@ def test_subset(self):
51
50
# with a subset of A should be the same
52
51
result = df .asof (dates , subset = 'A' )
53
52
expected = df .asof (dates )
54
- assert_frame_equal (result , expected )
53
+ tm . assert_frame_equal (result , expected )
55
54
56
55
# same with A/B
57
56
result = df .asof (dates , subset = ['A' , 'B' ])
58
57
expected = df .asof (dates )
59
- assert_frame_equal (result , expected )
58
+ tm . assert_frame_equal (result , expected )
60
59
61
60
# B gives self.df.asof
62
61
result = df .asof (dates , subset = 'B' )
63
62
expected = df .resample ('25s' , closed = 'right' ).ffill ().reindex (dates )
64
63
expected .iloc [20 :] = 9
65
64
66
- assert_frame_equal (result , expected )
65
+ tm . assert_frame_equal (result , expected )
67
66
68
67
def test_missing (self ):
69
68
# GH 15118
@@ -75,9 +74,15 @@ def test_missing(self):
75
74
result = df .asof ('1989-12-31' )
76
75
77
76
expected = Series (index = ['A' , 'B' ], name = Timestamp ('1989-12-31' ))
78
- assert_series_equal (result , expected )
77
+ tm . assert_series_equal (result , expected )
79
78
80
79
result = df .asof (to_datetime (['1989-12-31' ]))
81
80
expected = DataFrame (index = to_datetime (['1989-12-31' ]),
82
81
columns = ['A' , 'B' ], dtype = 'float64' )
83
- assert_frame_equal (result , expected )
82
+ tm .assert_frame_equal (result , expected )
83
+
84
+ def test_all_nans (self ):
85
+ # series is all nans
86
+ result = DataFrame ([np .nan ]).asof ([0 ])
87
+ expected = DataFrame ([np .nan ])
88
+ tm .assert_frame_equal (result , expected )
0 commit comments