Skip to content

Commit d5ea14a

Browse files
author
Maxime Leblanc
committed
Fix formatting in samples and tests
1 parent c6e0242 commit d5ea14a

2 files changed

Lines changed: 3 additions & 9 deletions

File tree

samples/browse_zones_darrets.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,7 @@ def main():
3838

3939
if args.search:
4040
search_lower = args.search.lower()
41-
records = [
42-
r for r in records if search_lower in json.dumps(r, ensure_ascii=False).lower()
43-
]
41+
records = [r for r in records if search_lower in json.dumps(r, ensure_ascii=False).lower()]
4442
print(f"Matching '{args.search}': {len(records)} records")
4543

4644
print(f"\nShowing first {min(args.limit, len(records))} records:\n")

tests/test_prim_api.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,7 @@ class TestLoadDataset:
6363
def test_load_dataset_reads_jsonl(self, tmp_path, monkeypatch):
6464
monkeypatch.setattr(ds_mod, "DATA_RAW_DIR", tmp_path)
6565
records = [{"id": 1, "name": "a"}, {"id": 2, "name": "b"}]
66-
(tmp_path / "test-ds.jsonl").write_text(
67-
"\n".join(json.dumps(r) for r in records) + "\n"
68-
)
66+
(tmp_path / "test-ds.jsonl").write_text("\n".join(json.dumps(r) for r in records) + "\n")
6967

7068
result = ds_mod.load_dataset("test-ds")
7169
assert result == records
@@ -178,9 +176,7 @@ def test_get_zones_darrets(self):
178176
patch.object(client_mod, "ApiClient"),
179177
patch.object(client_mod, "DefaultApi"),
180178
patch.object(client_mod, "ensure_all_datasets"),
181-
patch.object(
182-
client_mod, "load_dataset", return_value=[{"name": "Gare"}]
183-
) as mock_ld,
179+
patch.object(client_mod, "load_dataset", return_value=[{"name": "Gare"}]) as mock_ld,
184180
):
185181
sdk = client_mod.IdFMPrimAPI("my-key", auto_sync=False)
186182
result = sdk.get_zones_darrets()

0 commit comments

Comments
 (0)