|
2 | 2 | from datetime import date |
3 | 3 | from pathlib import Path |
4 | 4 |
|
5 | | -import numpy as np |
6 | 5 | import pandas as pd |
7 | 6 | import pytest |
8 | | - |
9 | 7 | from pysus.data.dbf_reader import ( |
10 | 8 | _parse_header, |
11 | 9 | read_dbf_fast, |
|
14 | 12 | stream_dbf_fast, |
15 | 13 | ) |
16 | 14 |
|
17 | | - |
18 | 15 | # --------------------------------------------------------------------------- |
19 | 16 | # Helpers |
20 | 17 | # --------------------------------------------------------------------------- |
@@ -85,9 +82,7 @@ def simple_dbf(tmp_dir): |
85 | 82 | def wide_dbf(tmp_dir): |
86 | 83 | dbf_path = tmp_dir / "wide.dbf" |
87 | 84 | fields = [(f"COL{i}", "C", 8, 0) for i in range(20)] |
88 | | - records = [ |
89 | | - tuple(f"val{i}_{j}" for j in range(20)) for i in range(100) |
90 | | - ] |
| 85 | + records = [tuple(f"val{i}_{j}" for j in range(20)) for i in range(100)] |
91 | 86 | _create_dbf(dbf_path, fields, records) |
92 | 87 | return dbf_path |
93 | 88 |
|
@@ -200,7 +195,6 @@ def test_read_fast_wide(wide_dbf): |
200 | 195 | def test_read_fast_strips_nul_bytes(tmp_dir): |
201 | 196 | dbf_path = tmp_dir / "nuls.dbf" |
202 | 197 | today = date.today() |
203 | | - fields = [("VAL", "C", 8, 0)] |
204 | 198 | buf = bytearray() |
205 | 199 | buf.append(0x03) |
206 | 200 | buf.append(today.year - 1900) |
@@ -289,15 +283,11 @@ def test_read_filtered_column_subset(disease_dbf): |
289 | 283 |
|
290 | 284 | def test_read_filtered_missing_column(disease_dbf): |
291 | 285 | with pytest.raises(KeyError, match="not found"): |
292 | | - read_dbf_filtered( |
293 | | - disease_dbf, column="NONEXISTENT", values=["X"] |
294 | | - ) |
| 286 | + read_dbf_filtered(disease_dbf, column="NONEXISTENT", values=["X"]) |
295 | 287 |
|
296 | 288 |
|
297 | 289 | def test_read_filtered_empty_file(empty_dbf): |
298 | | - df = read_dbf_filtered( |
299 | | - empty_dbf, column="NAME", values=["X"] |
300 | | - ) |
| 290 | + df = read_dbf_filtered(empty_dbf, column="NAME", values=["X"]) |
301 | 291 | assert len(df) == 0 |
302 | 292 |
|
303 | 293 |
|
@@ -378,9 +368,7 @@ def test_large_record_count(tmp_dir): |
378 | 368 |
|
379 | 369 |
|
380 | 370 | def test_field_name_lowercase_match(): |
381 | | - from pysus.data.dbf_reader import _find_field |
382 | | - |
383 | | - from pysus.data.dbf_reader import DBFSchema, DBFField |
| 371 | + from pysus.data.dbf_reader import DBFField, DBFSchema, _find_field |
384 | 372 |
|
385 | 373 | schema = DBFSchema( |
386 | 374 | num_records=1, |
|
0 commit comments