Skip to content

Commit f855954

Browse files
committed
last dict fixings
1 parent b5233c4 commit f855954

File tree

1 file changed

+59
-59
lines changed

1 file changed

+59
-59
lines changed

pandas/tests/io/generate_legacy_storage_files.py

Lines changed: 59 additions & 59 deletions
Original file line numberDiff line numberDiff line change
@@ -142,16 +142,16 @@ def create_data():
142142
"E": [0.0, 1, Timestamp("20100101"), "foo", 2.0],
143143
}
144144

145-
scalars = dict(timestamp=Timestamp("20130101"), period=Period("2012", "M"))
146-
147-
index = dict(
148-
int=Index(np.arange(10)),
149-
date=date_range("20130101", periods=10),
150-
period=period_range("2013-01-01", freq="M", periods=10),
151-
float=Index(np.arange(10, dtype=np.float64)),
152-
uint=Index(np.arange(10, dtype=np.uint64)),
153-
timedelta=timedelta_range("00:00:00", freq="30T", periods=10),
154-
)
145+
scalars = {"timestamp": Timestamp("20130101"), "period": Period("2012", "M")}
146+
147+
index = {
148+
"int": Index(np.arange(10)),
149+
"date": date_range("20130101", periods=10),
150+
"period": period_range("2013-01-01", freq="M", periods=10),
151+
"float": Index(np.arange(10, dtype=np.float64)),
152+
"uint": Index(np.arange(10, dtype=np.uint64)),
153+
"timedelta": timedelta_range("00:00:00", freq="30T", periods=10),
154+
}
155155

156156
index["range"] = RangeIndex(10)
157157

@@ -160,8 +160,8 @@ def create_data():
160160

161161
index["interval"] = interval_range(0, periods=10)
162162

163-
mi = dict(
164-
reg2=MultiIndex.from_tuples(
163+
mi = {
164+
"reg2": MultiIndex.from_tuples(
165165
tuple(
166166
zip(
167167
*[
@@ -172,35 +172,35 @@ def create_data():
172172
),
173173
names=["first", "second"],
174174
)
175-
)
175+
}
176176

177-
series = dict(
178-
float=Series(data["A"]),
179-
int=Series(data["B"]),
180-
mixed=Series(data["E"]),
181-
ts=Series(
177+
series = {
178+
"float": Series(data["A"]),
179+
"int": Series(data["B"]),
180+
"mixed": Series(data["E"]),
181+
"ts": Series(
182182
np.arange(10).astype(np.int64), index=date_range("20130101", periods=10)
183183
),
184-
mi=Series(
184+
"mi": Series(
185185
np.arange(5).astype(np.float64),
186186
index=MultiIndex.from_tuples(
187187
tuple(zip(*[[1, 1, 2, 2, 2], [3, 4, 3, 4, 5]])), names=["one", "two"]
188188
),
189189
),
190-
dup=Series(np.arange(5).astype(np.float64), index=["A", "B", "C", "D", "A"]),
191-
cat=Series(Categorical(["foo", "bar", "baz"])),
192-
dt=Series(date_range("20130101", periods=5)),
193-
dt_tz=Series(date_range("20130101", periods=5, tz="US/Eastern")),
194-
period=Series([Period("2000Q1")] * 5),
195-
)
190+
"dup": Series(np.arange(5).astype(np.float64), index=["A", "B", "C", "D", "A"]),
191+
"cat": Series(Categorical(["foo", "bar", "baz"])),
192+
"dt": Series(date_range("20130101", periods=5)),
193+
"dt_tz": Series(date_range("20130101", periods=5, tz="US/Eastern")),
194+
"period": Series([Period("2000Q1")] * 5),
195+
}
196196

197197
mixed_dup_df = DataFrame(data)
198198
mixed_dup_df.columns = list("ABCDA")
199-
frame = dict(
200-
float=DataFrame({"A": series["float"], "B": series["float"] + 1}),
201-
int=DataFrame({"A": series["int"], "B": series["int"] + 1}),
202-
mixed=DataFrame({k: data[k] for k in ["A", "B", "C", "D"]}),
203-
mi=DataFrame(
199+
frame = {
200+
"float": DataFrame({"A": series["float"], "B": series["float"] + 1}),
201+
"int": DataFrame({"A": series["int"], "B": series["int"] + 1}),
202+
"mixed": DataFrame({k: data[k] for k in ["A", "B", "C", "D"]}),
203+
"mi": DataFrame(
204204
{"A": np.arange(5).astype(np.float64), "B": np.arange(5).astype(np.int64)},
205205
index=MultiIndex.from_tuples(
206206
tuple(
@@ -214,45 +214,45 @@ def create_data():
214214
names=["first", "second"],
215215
),
216216
),
217-
dup=DataFrame(
217+
"dup": DataFrame(
218218
np.arange(15).reshape(5, 3).astype(np.float64), columns=["A", "B", "A"]
219219
),
220-
cat_onecol=DataFrame({"A": Categorical(["foo", "bar"])}),
221-
cat_and_float=DataFrame(
220+
"cat_onecol": DataFrame({"A": Categorical(["foo", "bar"])}),
221+
"cat_and_float": DataFrame(
222222
{
223223
"A": Categorical(["foo", "bar", "baz"]),
224224
"B": np.arange(3).astype(np.int64),
225225
}
226226
),
227-
mixed_dup=mixed_dup_df,
228-
dt_mixed_tzs=DataFrame(
227+
"mixed_dup": mixed_dup_df,
228+
"dt_mixed_tzs": DataFrame(
229229
{
230230
"A": Timestamp("20130102", tz="US/Eastern"),
231231
"B": Timestamp("20130603", tz="CET"),
232232
},
233233
index=range(5),
234234
),
235-
dt_mixed2_tzs=DataFrame(
235+
"dt_mixed2_tzs": DataFrame(
236236
{
237237
"A": Timestamp("20130102", tz="US/Eastern"),
238238
"B": Timestamp("20130603", tz="CET"),
239239
"C": Timestamp("20130603", tz="UTC"),
240240
},
241241
index=range(5),
242242
),
243-
)
243+
}
244244

245-
cat = dict(
246-
int8=Categorical(list("abcdefg")),
247-
int16=Categorical(np.arange(1000)),
248-
int32=Categorical(np.arange(10000)),
249-
)
245+
cat = {
246+
"int8": Categorical(list("abcdefg")),
247+
"int16": Categorical(np.arange(1000)),
248+
"int32": Categorical(np.arange(10000)),
249+
}
250250

251-
timestamp = dict(
252-
normal=Timestamp("2011-01-01"),
253-
nat=NaT,
254-
tz=Timestamp("2011-01-01", tz="US/Eastern"),
255-
)
251+
timestamp = {
252+
"normal": Timestamp("2011-01-01"),
253+
"nat": NaT,
254+
"tz": Timestamp("2011-01-01", tz="US/Eastern"),
255+
}
256256

257257
timestamp["freq"] = Timestamp("2011-01-01", freq="D")
258258
timestamp["both"] = Timestamp("2011-01-01", tz="Asia/Tokyo", freq="M")
@@ -282,18 +282,18 @@ def create_data():
282282
"Minute": Minute(1),
283283
}
284284

285-
return dict(
286-
series=series,
287-
frame=frame,
288-
index=index,
289-
scalars=scalars,
290-
mi=mi,
291-
sp_series=dict(float=_create_sp_series(), ts=_create_sp_tsseries()),
292-
sp_frame=dict(float=_create_sp_frame()),
293-
cat=cat,
294-
timestamp=timestamp,
295-
offsets=off,
296-
)
285+
return {
286+
"series": series,
287+
"frame": frame,
288+
"index": index,
289+
"scalars": scalars,
290+
"mi": mi,
291+
"sp_series": {"float": _create_sp_series(), "ts": _create_sp_tsseries()},
292+
"sp_frame": {"float": _create_sp_frame()},
293+
"cat": cat,
294+
"timestamp": timestamp,
295+
"offsets": off,
296+
}
297297

298298

299299
def create_pickle_data():

0 commit comments

Comments
 (0)