rzmk/czv
1
0
Fork 0
mirror of https://github.com/rzmk/czv.git synced 2025-12-19 08:09:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(python): rename tests & add column count test

Browse source
This commit is contained in:
rzmk 2024-06-25 21:19:24 -04:00
parent 1e7c4a1569
commit 5b8c057e5d
No known key found for this signature in database
1 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
czv-python/tests/test_count.py
View file

@ -2,12 +2,12 @@ import czv
import pytest
from .test_data import test_data
class TestCountFunc:
class TestRowCount:
@pytest.mark.parametrize(
"file_name,expected",
[("fruits.csv", 3), ("constituents_altnames.csv", 33971)],
)
def test_count(self, file_name, expected):
def test_row_count(self, file_name, expected):
"""Count the total number of non-header rows."""
result = czv.row_count(file_path=test_data[file_name])
@ -17,8 +17,19 @@ class TestCountFunc:
"file_name,expected",
[("fruits.csv", 4), ("constituents_altnames.csv", 33972)],
)
def test_include_header_row(self, file_name, expected):
def test_row_count_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
class TestColumnCount:
@pytest.mark.parametrize(
"file_name,expected",
[("fruits.csv", 2), ("constituents_altnames.csv", 6)],
)
def test_column_count(self, file_name, expected):
"""Count the total number of columns."""
result = czv.column_count(file_path=test_data[file_name])
assert result == expected