mirror of
https://github.com/rzmk/czv.git
synced 2025-12-19 08:09:24 +00:00
24 lines
807 B
Python
24 lines
807 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(test_data[file_name].read_text())
|
|
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(test_data[file_name].read_text(), include_header_row=True)
|
|
assert result == expected
|