Skip to content

Commit

Permalink
fixup! Add test for skip_first_n_rows
Browse files Browse the repository at this point in the history
  • Loading branch information
he3lixxx committed Dec 31, 2022
1 parent a3cb763 commit fb4b199
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions evap/staff/tests/test_importers.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from copy import deepcopy
from datetime import date, datetime
from typing import Iterable
from unittest.mock import patch

from django.core.exceptions import ValidationError
Expand All @@ -22,13 +21,11 @@


class TestExcelFileRowMapper(TestCase):
@dataclass
class SingleColumnInputRow(InputRow):
column_count = 1

def __init__(self, location: ExcelFileLocation, *cells: Iterable[str]):
assert len(cells) == 1
self.value = cells[0]
self.location = location
location: ExcelFileLocation
value: str

def test_skip_first_n_rows_handled_correctly(self):
workbook_data = {"SheetName": [[str(i)] for i in range(10)]}
Expand Down

0 comments on commit fb4b199

Please sign in to comment.