Skip to content

Commit

Permalink
test(python): rename tests & add column count test
Browse files Browse the repository at this point in the history
  • Loading branch information
rzmk committed Jun 26, 2024
1 parent 1e7c4a1 commit 5b8c057
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions czv-python/tests/test_count.py
Original file line number Diff line number Diff line change
@@ -2,12 +2,12 @@
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 @@ def test_count(self, file_name, expected):
"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

0 comments on commit 5b8c057

Please sign in to comment.