Skip to content

Commit fda70c0

Browse files
authored
STYLE: Inconsistent namespace - resample #39992 (#40220)
* Fix styling * Fix other linting issues * Revert
1 parent 73fbcc9 commit fda70c0

7 files changed

+82
-88
lines changed

pandas/tests/resample/test_datetime_index.py

+31-31
Original file line numberDiff line numberDiff line change
@@ -125,20 +125,20 @@ def test_resample_basic(series, closed, expected):
125125
def test_resample_integerarray():
126126
# GH 25580, resample on IntegerArray
127127
ts = Series(
128-
range(9), index=pd.date_range("1/1/2000", periods=9, freq="T"), dtype="Int64"
128+
range(9), index=date_range("1/1/2000", periods=9, freq="T"), dtype="Int64"
129129
)
130130
result = ts.resample("3T").sum()
131131
expected = Series(
132132
[3, 12, 21],
133-
index=pd.date_range("1/1/2000", periods=3, freq="3T"),
133+
index=date_range("1/1/2000", periods=3, freq="3T"),
134134
dtype="Int64",
135135
)
136136
tm.assert_series_equal(result, expected)
137137

138138
result = ts.resample("3T").mean()
139139
expected = Series(
140140
[1, 4, 7],
141-
index=pd.date_range("1/1/2000", periods=3, freq="3T"),
141+
index=date_range("1/1/2000", periods=3, freq="3T"),
142142
dtype="Float64",
143143
)
144144
tm.assert_series_equal(result, expected)
@@ -529,8 +529,8 @@ def test_resample_ohlc(series):
529529
def test_resample_ohlc_result():
530530

531531
# GH 12332
532-
index = pd.date_range("1-1-2000", "2-15-2000", freq="h")
533-
index = index.union(pd.date_range("4-15-2000", "5-15-2000", freq="h"))
532+
index = date_range("1-1-2000", "2-15-2000", freq="h")
533+
index = index.union(date_range("4-15-2000", "5-15-2000", freq="h"))
534534
s = Series(range(len(index)), index=index)
535535

536536
a = s.loc[:"4-15-2000"].resample("30T").ohlc()
@@ -805,7 +805,7 @@ def test_resample_bad_offset(offset):
805805
def test_resample_origin_prime_freq():
806806
# GH 31809
807807
start, end = "2000-10-01 23:30:00", "2000-10-02 00:30:00"
808-
rng = pd.date_range(start, end, freq="7min")
808+
rng = date_range(start, end, freq="7min")
809809
ts = Series(np.random.randn(len(rng)), index=rng)
810810

811811
exp_rng = date_range("2000-10-01 23:14:00", "2000-10-02 00:22:00", freq="17min")
@@ -863,7 +863,7 @@ def test_resample_origin_with_tz():
863863
def test_resample_origin_epoch_with_tz_day_vs_24h():
864864
# GH 34474
865865
start, end = "2000-10-01 23:30:00+0500", "2000-12-02 00:30:00+0500"
866-
rng = pd.date_range(start, end, freq="7min")
866+
rng = date_range(start, end, freq="7min")
867867
random_values = np.random.randn(len(rng))
868868
ts_1 = Series(random_values, index=rng)
869869

@@ -880,7 +880,7 @@ def test_resample_origin_epoch_with_tz_day_vs_24h():
880880

881881
# check that we have the similar results with two different timezones (+2H and +5H)
882882
start, end = "2000-10-01 23:30:00+0200", "2000-12-02 00:30:00+0200"
883-
rng = pd.date_range(start, end, freq="7min")
883+
rng = date_range(start, end, freq="7min")
884884
ts_2 = Series(random_values, index=rng)
885885
result_5 = ts_2.resample("D", origin="epoch").mean()
886886
result_6 = ts_2.resample("24H", origin="epoch").mean()
@@ -903,7 +903,7 @@ def _create_series(values, timestamps, freq="D"):
903903
# test classical behavior of origin in a DST context
904904
start = Timestamp("2013-11-02", tz=tz)
905905
end = Timestamp("2013-11-03 23:59", tz=tz)
906-
rng = pd.date_range(start, end, freq="1h")
906+
rng = date_range(start, end, freq="1h")
907907
ts = Series(np.ones(len(rng)), index=rng)
908908

909909
expected = _create_series([24.0, 25.0], ["2013-11-02", "2013-11-03"])
@@ -914,7 +914,7 @@ def _create_series(values, timestamps, freq="D"):
914914
# test complex behavior of origin/offset in a DST context
915915
start = Timestamp("2013-11-03", tz=tz)
916916
end = Timestamp("2013-11-03 23:59", tz=tz)
917-
rng = pd.date_range(start, end, freq="1h")
917+
rng = date_range(start, end, freq="1h")
918918
ts = Series(np.ones(len(rng)), index=rng)
919919

920920
expected_ts = ["2013-11-02 22:00-05:00", "2013-11-03 22:00-06:00"]
@@ -969,7 +969,7 @@ def test_period_with_agg():
969969
# aggregate a period resampler with a lambda
970970
s2 = Series(
971971
np.random.randint(0, 5, 50),
972-
index=pd.period_range("2012-01-01", freq="H", periods=50),
972+
index=period_range("2012-01-01", freq="H", periods=50),
973973
dtype="float64",
974974
)
975975

@@ -1003,7 +1003,7 @@ def test_resample_dtype_preservation():
10031003

10041004
df = DataFrame(
10051005
{
1006-
"date": pd.date_range(start="2016-01-01", periods=4, freq="W"),
1006+
"date": date_range(start="2016-01-01", periods=4, freq="W"),
10071007
"group": [1, 1, 2, 2],
10081008
"val": Series([5, 6, 7, 8], dtype="int32"),
10091009
}
@@ -1022,7 +1022,7 @@ def test_resample_dtype_coercion():
10221022

10231023
# GH 16361
10241024
df = {"a": [1, 3, 1, 4]}
1025-
df = DataFrame(df, index=pd.date_range("2017-01-01", "2017-01-04"))
1025+
df = DataFrame(df, index=date_range("2017-01-01", "2017-01-04"))
10261026

10271027
expected = df.astype("float64").resample("H").mean()["a"].interpolate("cubic")
10281028

@@ -1056,12 +1056,12 @@ def test_nanosecond_resample_error():
10561056
# Resampling using pd.tseries.offsets.Nano as period
10571057
start = 1443707890427
10581058
exp_start = 1443707890400
1059-
indx = pd.date_range(start=pd.to_datetime(start), periods=10, freq="100n")
1059+
indx = date_range(start=pd.to_datetime(start), periods=10, freq="100n")
10601060
ts = Series(range(len(indx)), index=indx)
10611061
r = ts.resample(pd.tseries.offsets.Nano(100))
10621062
result = r.agg("mean")
10631063

1064-
exp_indx = pd.date_range(start=pd.to_datetime(exp_start), periods=10, freq="100n")
1064+
exp_indx = date_range(start=pd.to_datetime(exp_start), periods=10, freq="100n")
10651065
exp = Series(range(len(exp_indx)), index=exp_indx)
10661066

10671067
tm.assert_series_equal(result, exp)
@@ -1128,8 +1128,8 @@ def test_resample_anchored_multiday():
11281128
#
11291129
# See: https://github.com/pandas-dev/pandas/issues/8683
11301130

1131-
index1 = pd.date_range("2014-10-14 23:06:23.206", periods=3, freq="400L")
1132-
index2 = pd.date_range("2014-10-15 23:00:00", periods=2, freq="2200L")
1131+
index1 = date_range("2014-10-14 23:06:23.206", periods=3, freq="400L")
1132+
index2 = date_range("2014-10-15 23:00:00", periods=2, freq="2200L")
11331133
index = index1.union(index2)
11341134

11351135
s = Series(np.random.randn(5), index=index)
@@ -1174,7 +1174,7 @@ def test_anchored_lowercase_buglet():
11741174

11751175
def test_upsample_apply_functions():
11761176
# #1596
1177-
rng = pd.date_range("2012-06-12", periods=4, freq="h")
1177+
rng = date_range("2012-06-12", periods=4, freq="h")
11781178

11791179
ts = Series(np.random.randn(len(rng)), index=rng)
11801180

@@ -1183,7 +1183,7 @@ def test_upsample_apply_functions():
11831183

11841184

11851185
def test_resample_not_monotonic():
1186-
rng = pd.date_range("2012-06-12", periods=200, freq="h")
1186+
rng = date_range("2012-06-12", periods=200, freq="h")
11871187
ts = Series(np.random.randn(len(rng)), index=rng)
11881188

11891189
ts = ts.take(np.random.permutation(len(ts)))
@@ -1255,12 +1255,12 @@ def test_resample_consistency():
12551255
# GH 6418
12561256
# resample with bfill / limit / reindex consistency
12571257

1258-
i30 = pd.date_range("2002-02-02", periods=4, freq="30T")
1258+
i30 = date_range("2002-02-02", periods=4, freq="30T")
12591259
s = Series(np.arange(4.0), index=i30)
12601260
s[2] = np.NaN
12611261

12621262
# Upsample by factor 3 with reindex() and resample() methods:
1263-
i10 = pd.date_range(i30[0], i30[-1], freq="10T")
1263+
i10 = date_range(i30[0], i30[-1], freq="10T")
12641264

12651265
s10 = s.reindex(index=i10, method="bfill")
12661266
s10_2 = s.reindex(index=i10, method="bfill", limit=2)
@@ -1364,8 +1364,8 @@ def test_resample_nunique_preserves_column_level_names():
13641364

13651365
def test_resample_nunique_with_date_gap():
13661366
# GH 13453
1367-
index = pd.date_range("1-1-2000", "2-15-2000", freq="h")
1368-
index2 = pd.date_range("4-15-2000", "5-15-2000", freq="h")
1367+
index = date_range("1-1-2000", "2-15-2000", freq="h")
1368+
index2 = date_range("4-15-2000", "5-15-2000", freq="h")
13691369
index3 = index.append(index2)
13701370
s = Series(range(len(index3)), index=index3, dtype="int64")
13711371
r = s.resample("M")
@@ -1462,7 +1462,7 @@ def test_groupby_with_dst_time_change():
14621462

14631463
df = DataFrame([1, 2], index=index)
14641464
result = df.groupby(Grouper(freq="1d")).last()
1465-
expected_index_values = pd.date_range(
1465+
expected_index_values = date_range(
14661466
"2016-11-02", "2016-11-24", freq="d", tz="America/Chicago"
14671467
)
14681468

@@ -1587,11 +1587,11 @@ def test_downsample_across_dst_weekly():
15871587
)
15881588
tm.assert_frame_equal(result, expected)
15891589

1590-
idx = pd.date_range("2013-04-01", "2013-05-01", tz="Europe/London", freq="H")
1590+
idx = date_range("2013-04-01", "2013-05-01", tz="Europe/London", freq="H")
15911591
s = Series(index=idx, dtype=np.float64)
15921592
result = s.resample("W").mean()
15931593
expected = Series(
1594-
index=pd.date_range("2013-04-07", freq="W", periods=5, tz="Europe/London"),
1594+
index=date_range("2013-04-07", freq="W", periods=5, tz="Europe/London"),
15951595
dtype=np.float64,
15961596
)
15971597
tm.assert_series_equal(result, expected)
@@ -1601,7 +1601,7 @@ def test_downsample_dst_at_midnight():
16011601
# GH 25758
16021602
start = datetime(2018, 11, 3, 12)
16031603
end = datetime(2018, 11, 5, 12)
1604-
index = pd.date_range(start, end, freq="1H")
1604+
index = date_range(start, end, freq="1H")
16051605
index = index.tz_localize("UTC").tz_convert("America/Havana")
16061606
data = list(range(len(index)))
16071607
dataframe = DataFrame(data, index=index)
@@ -1681,7 +1681,7 @@ def f(data, add_arg):
16811681
tm.assert_series_equal(result, expected)
16821682

16831683
# Testing dataframe
1684-
df = DataFrame({"A": 1, "B": 2}, index=pd.date_range("2017", periods=10))
1684+
df = DataFrame({"A": 1, "B": 2}, index=date_range("2017", periods=10))
16851685
result = df.groupby("A").resample("D").agg(f, multiplier)
16861686
expected = df.groupby("A").resample("D").mean().multiply(multiplier)
16871687
tm.assert_frame_equal(result, expected)
@@ -1707,7 +1707,7 @@ def test_resample_equivalent_offsets(n1, freq1, n2, freq2, k):
17071707
# GH 24127
17081708
n1_ = n1 * k
17091709
n2_ = n2 * k
1710-
s = Series(0, index=pd.date_range("19910905 13:00", "19911005 07:00", freq=freq1))
1710+
s = Series(0, index=date_range("19910905 13:00", "19911005 07:00", freq=freq1))
17111711
s = s + range(len(s))
17121712

17131713
result1 = s.resample(str(n1_) + freq1).mean()
@@ -1797,10 +1797,10 @@ def test_resample_calendar_day_with_dst(
17971797
first: str, last: str, freq_in: str, freq_out: str, exp_last: str
17981798
):
17991799
# GH 35219
1800-
ts = Series(1.0, pd.date_range(first, last, freq=freq_in, tz="Europe/Amsterdam"))
1800+
ts = Series(1.0, date_range(first, last, freq=freq_in, tz="Europe/Amsterdam"))
18011801
result = ts.resample(freq_out).pad()
18021802
expected = Series(
1803-
1.0, pd.date_range(first, exp_last, freq=freq_out, tz="Europe/Amsterdam")
1803+
1.0, date_range(first, exp_last, freq=freq_out, tz="Europe/Amsterdam")
18041804
)
18051805
tm.assert_series_equal(result, expected)
18061806

pandas/tests/resample/test_deprecated.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def _create_index(*args, **kwargs):
5252
def test_deprecating_on_loffset_and_base():
5353
# GH 31809
5454

55-
idx = pd.date_range("2001-01-01", periods=4, freq="T")
55+
idx = date_range("2001-01-01", periods=4, freq="T")
5656
df = DataFrame(data=4 * [range(2)], index=idx, columns=["a", "b"])
5757

5858
with tm.assert_produces_warning(FutureWarning):
@@ -243,7 +243,7 @@ def test_loffset_returns_datetimeindex(frame, kind, agg_arg):
243243
)
244244
def test_resample_with_non_zero_base(start, end, start_freq, end_freq, base, offset):
245245
# GH 23882
246-
s = Series(0, index=pd.period_range(start, end, freq=start_freq))
246+
s = Series(0, index=period_range(start, end, freq=start_freq))
247247
s = s + np.arange(len(s))
248248
with tm.assert_produces_warning(FutureWarning):
249249
result = s.resample(end_freq, base=base).mean()

pandas/tests/resample/test_period_index.py

+12-14
Original file line numberDiff line numberDiff line change
@@ -224,9 +224,9 @@ def test_resample_basic(self):
224224
)
225225
def test_resample_count(self, freq, expected_vals):
226226
# GH12774
227-
series = Series(1, index=pd.period_range(start="2000", periods=100))
227+
series = Series(1, index=period_range(start="2000", periods=100))
228228
result = series.resample(freq).count()
229-
expected_index = pd.period_range(
229+
expected_index = period_range(
230230
start="2000", freq=freq, periods=len(expected_vals)
231231
)
232232
expected = Series(expected_vals, index=expected_index)
@@ -235,9 +235,7 @@ def test_resample_count(self, freq, expected_vals):
235235
def test_resample_same_freq(self, resample_method):
236236

237237
# GH12770
238-
series = Series(
239-
range(3), index=pd.period_range(start="2000", periods=3, freq="M")
240-
)
238+
series = Series(range(3), index=period_range(start="2000", periods=3, freq="M"))
241239
expected = series
242240

243241
result = getattr(series.resample("M"), resample_method)()
@@ -250,7 +248,7 @@ def test_resample_incompat_freq(self):
250248
)
251249
with pytest.raises(IncompatibleFrequency, match=msg):
252250
Series(
253-
range(3), index=pd.period_range(start="2000", periods=3, freq="M")
251+
range(3), index=period_range(start="2000", periods=3, freq="M")
254252
).resample("W").mean()
255253

256254
def test_with_local_timezone_pytz(self):
@@ -261,7 +259,7 @@ def test_with_local_timezone_pytz(self):
261259
# 1 day later
262260
end = datetime(year=2013, month=11, day=2, hour=0, minute=0, tzinfo=pytz.utc)
263261

264-
index = pd.date_range(start, end, freq="H")
262+
index = date_range(start, end, freq="H")
265263

266264
series = Series(1, index=index)
267265
series = series.tz_convert(local_timezone)
@@ -270,14 +268,14 @@ def test_with_local_timezone_pytz(self):
270268
# Create the expected series
271269
# Index is moved back a day with the timezone conversion from UTC to
272270
# Pacific
273-
expected_index = pd.period_range(start=start, end=end, freq="D") - offsets.Day()
271+
expected_index = period_range(start=start, end=end, freq="D") - offsets.Day()
274272
expected = Series(1, index=expected_index)
275273
tm.assert_series_equal(result, expected)
276274

277275
def test_resample_with_pytz(self):
278276
# GH 13238
279277
s = Series(
280-
2, index=pd.date_range("2017-01-01", periods=48, freq="H", tz="US/Eastern")
278+
2, index=date_range("2017-01-01", periods=48, freq="H", tz="US/Eastern")
281279
)
282280
result = s.resample("D").mean()
283281
expected = Series(
@@ -302,7 +300,7 @@ def test_with_local_timezone_dateutil(self):
302300
year=2013, month=11, day=2, hour=0, minute=0, tzinfo=dateutil.tz.tzutc()
303301
)
304302

305-
index = pd.date_range(start, end, freq="H", name="idx")
303+
index = date_range(start, end, freq="H", name="idx")
306304

307305
series = Series(1, index=index)
308306
series = series.tz_convert(local_timezone)
@@ -312,7 +310,7 @@ def test_with_local_timezone_dateutil(self):
312310
# Index is moved back a day with the timezone conversion from UTC to
313311
# Pacific
314312
expected_index = (
315-
pd.period_range(start=start, end=end, freq="D", name="idx") - offsets.Day()
313+
period_range(start=start, end=end, freq="D", name="idx") - offsets.Day()
316314
)
317315
expected = Series(1, index=expected_index)
318316
tm.assert_series_equal(result, expected)
@@ -343,7 +341,7 @@ def test_resample_nonexistent_time_bin_edge(self):
343341

344342
def test_resample_ambiguous_time_bin_edge(self):
345343
# GH 10117
346-
idx = pd.date_range(
344+
idx = date_range(
347345
"2014-10-25 22:00:00", "2014-10-26 00:30:00", freq="30T", tz="Europe/London"
348346
)
349347
expected = Series(np.zeros(len(idx)), index=idx)
@@ -827,7 +825,7 @@ def test_resample_with_only_nat(self):
827825
)
828826
def test_resample_with_offset(self, start, end, start_freq, end_freq, offset):
829827
# GH 23882 & 31809
830-
s = Series(0, index=pd.period_range(start, end, freq=start_freq))
828+
s = Series(0, index=period_range(start, end, freq=start_freq))
831829
s = s + np.arange(len(s))
832830
result = s.resample(end_freq, offset=offset).mean()
833831
result = result.to_timestamp(end_freq)
@@ -869,7 +867,7 @@ def test_get_period_range_edges(self, first, last, freq, exp_first, exp_last):
869867

870868
def test_sum_min_count(self):
871869
# GH 19974
872-
index = pd.date_range(start="2018", freq="M", periods=6)
870+
index = date_range(start="2018", freq="M", periods=6)
873871
data = np.ones(6)
874872
data[3:6] = np.nan
875873
s = Series(data, index).to_period()

0 commit comments

Comments
 (0)