File tree Expand file tree Collapse file tree 2 files changed +26
-10
lines changed Expand file tree Collapse file tree 2 files changed +26
-10
lines changed Original file line number Diff line number Diff line change @@ -1882,15 +1882,20 @@ def test_add_new_column_infer_string():
1882
1882
1883
1883
def test_datetime_indexer_consistency_pyarrow_date32 ():
1884
1884
# GH#62158
1885
+ pytest .importorskip ("pyarrow" , minversion = "13.0.0" )
1885
1886
ser = Series (["2016-01-01" ], dtype = "date32[pyarrow]" )
1886
1887
ser3 = ser .astype ("datetime64[ns]" )
1887
1888
dti = Index (ser3 )
1888
- # All should be consistent
1889
- assert dti .get_loc (ser [0 ]) == 0
1890
- tm .assert_numpy_array_equal (dti .get_indexer (ser .values ), np .array ([0 ]))
1891
- tm .assert_numpy_array_equal (
1892
- dti .get_indexer (ser .values .astype (object )), np .array ([0 ])
1893
- )
1889
+
1890
+ with pytest .raises (KeyError ):
1891
+ dti .get_loc (ser [0 ])
1892
+
1893
+ # get_indexer returns -1 for both Arrow array and object-cast
1894
+ result = dti .get_indexer (ser .values )
1895
+ tm .assert_numpy_array_equal (result , np .array ([- 1 ], dtype = np .intp ))
1896
+
1897
+ result_obj = dti .get_indexer (ser .values .astype (object ))
1898
+ tm .assert_numpy_array_equal (result_obj , np .array ([- 1 ], dtype = np .intp ))
1894
1899
1895
1900
1896
1901
class TestSetitemValidation :
Original file line number Diff line number Diff line change @@ -767,10 +767,21 @@ def test_align_date_objects_with_datetimeindex(self):
767
767
768
768
result = ts + ts2
769
769
result2 = ts2 + ts
770
- expected = ts + ts [5 :]
771
- expected .index = expected .index ._with_freq (None )
772
- tm .assert_series_equal (result , expected )
773
- tm .assert_series_equal (result2 , expected )
770
+
771
+ date_labels = [x .date () for x in rng [5 :]]
772
+ expected_index = Index (list (rng ) + date_labels , dtype = object )
773
+
774
+ # Length and index checks
775
+ assert len (result ) == 35
776
+ tm .assert_index_equal (result .index , expected_index )
777
+ tm .assert_index_equal (result2 .index , expected_index )
778
+ assert result .index .dtype == object
779
+
780
+ # All NaN because there are no matching labels now
781
+ assert result .isna ().all ()
782
+ assert result2 .isna ().all ()
783
+
784
+ tm .assert_series_equal (result , result2 )
774
785
775
786
776
787
class TestNamePreservation :
You can’t perform that action at this time.
0 commit comments