diff --git a/klausur-service/backend/tests/test_cv_vocab_pipeline.py b/klausur-service/backend/tests/test_cv_vocab_pipeline.py index 4e79929..603509c 100644 --- a/klausur-service/backend/tests/test_cv_vocab_pipeline.py +++ b/klausur-service/backend/tests/test_cv_vocab_pipeline.py @@ -1327,8 +1327,11 @@ class TestCellsToVocabEntriesPageRef: 'ocr_engine': 'tesseract', }, ] + columns_meta = [ + {'type': 'column_en'}, {'type': 'column_de'}, {'type': 'page_ref'}, + ] - entries = _cells_to_vocab_entries(cells) + entries = _cells_to_vocab_entries(cells, columns_meta) assert len(entries) == 1 assert entries[0]['english'] == 'hello' @@ -1350,8 +1353,9 @@ class TestCellsToVocabEntriesPageRef: 'ocr_engine': 'tesseract', }, ] + columns_meta = [{'type': 'column_en'}] - entries = _cells_to_vocab_entries(cells) + entries = _cells_to_vocab_entries(cells, columns_meta) assert len(entries) == 1 assert entries[0]['source_page'] == ''