Skip to content

Commit 6835800

Browse files
committed
update tests
1 parent 21de024 commit 6835800

File tree

13 files changed

+39
-31
lines changed

13 files changed

+39
-31
lines changed

pandas/core/dtypes/cast.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1191,7 +1191,7 @@ def maybe_infer_to_datetimelike(
11911191
# numpy would have done it for us.
11921192
convert_numeric=False,
11931193
convert_non_numeric=True,
1194-
dtype_if_all_nat=np.dtype("M8[ns]"),
1194+
dtype_if_all_nat=np.dtype("M8[s]"),
11951195
)
11961196

11971197

pandas/tests/dtypes/test_dtypes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,8 @@ def test_is_boolean(self, categories, expected):
200200

201201
def test_dtype_specific_categorical_dtype(self):
202202
expected = "datetime64[ns]"
203-
result = str(Categorical(DatetimeIndex([])).categories.dtype)
203+
dti = DatetimeIndex([], dtype=expected)
204+
result = str(Categorical(dti).categories.dtype)
204205
assert result == expected
205206

206207
def test_not_string(self):

pandas/tests/frame/methods/test_transpose.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ def test_transpose_td64_intervals(self):
2929

3030
def test_transpose_empty_preserves_datetimeindex(self):
3131
# GH#41382
32-
df = DataFrame(index=DatetimeIndex([]))
32+
dti = DatetimeIndex([], dtype="M8[ns]")
33+
df = DataFrame(index=dti)
3334

3435
expected = DatetimeIndex([], dtype="datetime64[ns]", freq=None)
3536

pandas/tests/indexes/datetimes/test_date_range.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,7 @@ def test_date_range_edges(self, freq):
184184
exp = DatetimeIndex(
185185
[ts + n * td for n in range(1, 5)],
186186
freq=freq,
187+
dtype="M8[ns]",
187188
)
188189
tm.assert_index_equal(idx, exp)
189190

@@ -193,7 +194,7 @@ def test_date_range_edges(self, freq):
193194
end=ts + td,
194195
freq=freq,
195196
)
196-
exp = DatetimeIndex([], freq=freq)
197+
exp = DatetimeIndex([], dtype="M8[ns]", freq=freq)
197198
tm.assert_index_equal(idx, exp)
198199

199200
# start matches end
@@ -202,7 +203,7 @@ def test_date_range_edges(self, freq):
202203
end=ts + td,
203204
freq=freq,
204205
)
205-
exp = DatetimeIndex([ts + td], freq=freq)
206+
exp = DatetimeIndex([ts + td], dtype="M8[ns]", freq=freq)
206207
tm.assert_index_equal(idx, exp)
207208

208209
def test_date_range_near_implementation_bound(self):

pandas/tests/indexes/datetimes/test_scalar_compat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,7 @@ def test_dti_fields(self, tz):
304304
exp = dti[[0, 90, 181, 273]]
305305
tm.assert_index_equal(res, exp)
306306
res = dti[dti.is_leap_year]
307-
exp = DatetimeIndex([], freq="D", tz=dti.tz, name="name")
307+
exp = DatetimeIndex([], freq="D", tz=dti.tz, name="name").as_unit("ns")
308308
tm.assert_index_equal(res, exp)
309309

310310
def test_dti_is_year_quarter_start(self):

pandas/tests/indexes/datetimes/test_setops.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def test_union(self, tz, sort):
7373
expected2_notsorted = DatetimeIndex(list(other2) + list(rng2[:3]))
7474

7575
rng3 = date_range("1/1/2000", freq="D", periods=5, tz=tz)
76-
other3 = DatetimeIndex([], tz=tz)
76+
other3 = DatetimeIndex([], tz=tz).as_unit("ns")
7777
expected3 = date_range("1/1/2000", freq="D", periods=5, tz=tz)
7878
expected3_notsorted = rng3
7979

@@ -235,7 +235,7 @@ def test_intersection(self, tz, sort):
235235
expected3 = date_range("6/1/2000", "6/20/2000", freq="D", name=None)
236236

237237
rng4 = date_range("7/1/2000", "7/31/2000", freq="D", name="idx")
238-
expected4 = DatetimeIndex([], freq="D", name="idx")
238+
expected4 = DatetimeIndex([], freq="D", name="idx", dtype="M8[ns]")
239239

240240
for rng, expected in [
241241
(rng2, expected2),
@@ -269,7 +269,7 @@ def test_intersection(self, tz, sort):
269269

270270
# GH 7880
271271
rng4 = date_range("7/1/2000", "7/31/2000", freq="D", tz=tz, name="idx")
272-
expected4 = DatetimeIndex([], tz=tz, name="idx")
272+
expected4 = DatetimeIndex([], tz=tz, name="idx").as_unit("ns")
273273
assert expected4.freq is None
274274

275275
for rng, expected in [
@@ -540,7 +540,7 @@ def test_intersection(self):
540540

541541
# non-overlapping
542542
the_int = rng[:10].intersection(rng[10:])
543-
expected = DatetimeIndex([])
543+
expected = DatetimeIndex([]).as_unit("ns")
544544
tm.assert_index_equal(the_int, expected)
545545

546546
def test_intersection_bug(self):

pandas/tests/indexes/interval/test_interval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -388,7 +388,7 @@ def test_maybe_convert_i8_nat(self, breaks):
388388
# GH 20636
389389
index = IntervalIndex.from_breaks(breaks)
390390

391-
to_convert = breaks._constructor([pd.NaT] * 3)
391+
to_convert = breaks._constructor([pd.NaT] * 3).as_unit("ns")
392392
expected = Index([np.nan] * 3, dtype=np.float64)
393393
result = index._maybe_convert_i8(to_convert)
394394
tm.assert_index_equal(result, expected)

pandas/tests/indexes/period/test_partial_slicing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def test_range_slice_outofbounds(self, make_range):
117117
idx = make_range(start="2013/10/01", freq="D", periods=10)
118118

119119
df = DataFrame({"units": [100 + i for i in range(10)]}, index=idx)
120-
empty = DataFrame(index=type(idx)([], freq="D"), columns=["units"])
120+
empty = DataFrame(index=idx[:0], columns=["units"])
121121
empty["units"] = empty["units"].astype("int64")
122122

123123
tm.assert_frame_equal(df["2013/09/01":"2013/09/30"], empty)

pandas/tests/indexes/test_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def test_constructor_int_dtype_nan(self):
176176
"klass,dtype,na_val",
177177
[
178178
(Index, np.float64, np.nan),
179-
(DatetimeIndex, "datetime64[ns]", pd.NaT),
179+
(DatetimeIndex, "datetime64[s]", pd.NaT),
180180
],
181181
)
182182
def test_index_ctor_infer_nan_nat(self, klass, dtype, na_val):

pandas/tests/indexes/test_index_new.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,9 @@ def test_constructor_infer_nat_dt_like(
141141
)
142142

143143
expected = klass([NaT, NaT])
144+
if dtype[0] == "d":
145+
# we infer all-NaT as second resolution
146+
expected = expected.astype("M8[ns]")
144147
assert expected.dtype == dtype
145148
data = [ctor]
146149
data.insert(pos, nulls_fixture)

0 commit comments

Comments
 (0)