Skip to content

Commit 18adba3

Browse files
committed
fixed linting
1 parent e13cf26 commit 18adba3

File tree

3 files changed

+21
-27
lines changed

3 files changed

+21
-27
lines changed

pandas/tests/frame/test_query_eval.py

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -160,21 +160,13 @@ def test_query_empty_string(self):
160160
df.query("")
161161

162162
def test_query_duplicate_column_name(self, engine, parser):
163-
df = DataFrame(
164-
{
165-
"A": range(3),
166-
"B": range(3),
167-
"C": range(3)
168-
}
169-
).rename(columns={"B": "A"})
163+
df = DataFrame({"A": range(3), "B": range(3), "C": range(3)}).rename(
164+
columns={"B": "A"}
165+
)
170166

171-
res = df.query('C == 1', engine=engine, parser=parser)
167+
res = df.query("C == 1", engine=engine, parser=parser)
172168

173-
expect = DataFrame(
174-
[[1, 1, 1]],
175-
columns=["A", "A", "C"],
176-
index=[1]
177-
)
169+
expect = DataFrame([[1, 1, 1]], columns=["A", "A", "C"], index=[1])
178170

179171
tm.assert_frame_equal(res, expect)
180172

@@ -1141,9 +1133,7 @@ def test_query_with_nested_special_character(self, parser, engine):
11411133
[">=", operator.ge],
11421134
],
11431135
)
1144-
def test_query_lex_compare_strings(
1145-
self, parser, engine, op, func
1146-
):
1136+
def test_query_lex_compare_strings(self, parser, engine, op, func):
11471137
a = Series(np.random.default_rng(2).choice(list("abcde"), 20))
11481138
b = Series(np.arange(a.size))
11491139
df = DataFrame({"X": a, "Y": b})

pandas/tests/indexes/categorical/test_formats.py

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -51,27 +51,31 @@ def test_string_categorical_index_repr(self):
5151

5252
# multiple lines
5353
idx = CategoricalIndex(["あ", "いい", "ううう"] * 10)
54-
expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
55-
'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
56-
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
57-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""egory')""" # noqa: E501
54+
expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい',
55+
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
56+
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
57+
'ううう', 'あ', 'いい', 'ううう', 'あ',
58+
'いい', 'ううう', 'あ', 'いい', 'ううう'],
59+
categories=['あ', 'いい', 'ううう'],
60+
ordered=False, dtype='category')"""
5861

5962
assert repr(idx) == expected
6063

6164
# truncated
6265
idx = CategoricalIndex(["あ", "いい", "ううう"] * 100)
6366
expected = """CategoricalIndex(['あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
6467
...
65-
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
66-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""length=300)""" # noqa: E501
68+
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ',
69+
'いい', 'ううう'],
70+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""
6771

6872
assert repr(idx) == expected
6973

7074
# larger categories
7175
idx = CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
7276
expected = """CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ', 'さ', 'し',
7377
'す', 'せ', 'そ'],
74-
categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')"""ory')""" # noqa: E501
78+
categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')"""
7579

7680
assert repr(idx) == expected
7781

@@ -88,7 +92,7 @@ def test_string_categorical_index_repr(self):
8892
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
8993
'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい',
9094
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう'],
91-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""ory')""" # noqa: E501
95+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category')"""
9296

9397
assert repr(idx) == expected
9498

@@ -99,14 +103,14 @@ def test_string_categorical_index_repr(self):
99103
...
100104
'ううう', 'あ', 'いい', 'ううう', 'あ', 'いい', 'ううう',
101105
'あ', 'いい', 'ううう'],
102-
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""ength=300)""" # noqa: E501
106+
categories=['あ', 'いい', 'ううう'], ordered=False, dtype='category', length=300)"""
103107

104108
assert repr(idx) == expected
105109

106110
# larger categories
107111
idx = CategoricalIndex(list("あいうえおかきくけこさしすせそ"))
108112
expected = """CategoricalIndex(['あ', 'い', 'う', 'え', 'お', 'か', 'き', 'く', 'け', 'こ',
109113
'さ', 'し', 'す', 'せ', 'そ'],
110-
categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')""")""" # noqa: E501
114+
categories=['あ', 'い', 'う', 'え', ..., 'し', 'す', 'せ', 'そ'], ordered=False, dtype='category')"""
111115

112116
assert repr(idx) == expected

pandas/tests/io/parser/usecols/test_strings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def test_usecols_with_multi_byte_characters(all_parsers):
7979
data = """あああ,いい,ううう,ええええ
8080
0.056674973,8,True,a
8181
2.613230982,2,False,b
82-
3.568935038,7,False,a"""e,a"""
82+
3.568935038,7,False,a"""
8383
parser = all_parsers
8484

8585
exp_data = {

0 commit comments

Comments
 (0)