Skip to content

Commit fc4993f

Browse files
committed
ENH solve linting via black8 for Add nrow parameter for line delimited json for read_json #33916
1 parent 15d1d1e commit fc4993f

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

pandas/io/json/_json.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -645,7 +645,7 @@ def __init__(
645645
lines,
646646
chunksize,
647647
compression,
648-
nrows
648+
nrows,
649649
):
650650

651651
self.path_or_buf = filepath_or_buffer
@@ -742,7 +742,7 @@ def read(self):
742742
data = ensure_str(self.data)
743743
data = data.split("\n")
744744
if self.nrows:
745-
data = data[:self.nrows]
745+
data = data[: self.nrows]
746746
obj = self._get_object_parser(self._combine_lines(data))
747747
else:
748748
obj = self._get_object_parser(self.data)

pandas/tests/io/json/test_readlines.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def test_readjson_chunks_closes(chunksize):
130130
lines=True,
131131
chunksize=chunksize,
132132
compression=None,
133-
nrows=None
133+
nrows=None,
134134
)
135135
reader.read()
136136
assert (
@@ -185,10 +185,10 @@ def test_readjson_unicode(monkeypatch):
185185
@pytest.mark.parametrize("nrows", [1, 2])
186186
def test_readjson_nrows(nrows):
187187
# Test reading line-format JSON to Series with nrows param
188-
jsonl = '''{"a": 1, "b": 2}
188+
jsonl = """{"a": 1, "b": 2}
189189
{"a": 3, "b": 4}
190190
{"a": 5, "b": 6}
191-
{"a": 7, "b": 8}'''
191+
{"a": 7, "b": 8}"""
192192
result = pd.read_json(jsonl, lines=True, nrows=nrows)
193193
expected = pd.DataFrame({"a": [1, 3, 5, 7], "b": [2, 4, 6, 8]}).iloc[:nrows]
194194
tm.assert_frame_equal(result, expected)
@@ -197,10 +197,10 @@ def test_readjson_nrows(nrows):
197197
@pytest.mark.parametrize("nrows,chunksize", [(2, 2), (4, 2)])
198198
def test_readjson_nrows_chunks(nrows, chunksize):
199199
# Test reading line-format JSON to Series with nrows and chunksize param
200-
jsonl = '''{"a": 1, "b": 2}
200+
jsonl = """{"a": 1, "b": 2}
201201
{"a": 3, "b": 4}
202202
{"a": 5, "b": 6}
203-
{"a": 7, "b": 8}'''
203+
{"a": 7, "b": 8}"""
204204
reader = read_json(jsonl, lines=True, nrows=nrows, chunksize=chunksize)
205205
chunked = pd.concat(reader)
206206
expected = pd.DataFrame({"a": [1, 3, 5, 7], "b": [2, 4, 6, 8]}).iloc[:nrows]

0 commit comments

Comments
 (0)