mirror of
https://github.com/rzmk/czv.git
synced 2025-12-19 08:09:24 +00:00
ce260e9491
- Add `file_path` for czv-python count operations - Refactor tests based on file path instead of data - Use thiserror in czv - Update examples to reflect changes
24 lines
803 B
Python
24 lines
803 B
Python
import czv
|
|
import pytest
|
|
from .test_data import test_data
|
|
|
|
class TestCountFunc:
|
|
@pytest.mark.parametrize(
|
|
"file_name,expected",
|
|
[("fruits.csv", 3), ("constituents_altnames.csv", 33971)],
|
|
)
|
|
def test_count(self, file_name, expected):
|
|
"""Count the total number of non-header rows."""
|
|
|
|
result = czv.row_count(file_path=test_data[file_name])
|
|
assert result == expected
|
|
|
|
@pytest.mark.parametrize(
|
|
"file_name,expected",
|
|
[("fruits.csv", 4), ("constituents_altnames.csv", 33972)],
|
|
)
|
|
def test_include_header_row(self, file_name, expected):
|
|
"""Count the total number of rows including the header row."""
|
|
|
|
result = czv.row_count(file_path=test_data[file_name], include_header_row=True)
|
|
assert result == expected
|