Skip to content

Commit

Permalink
test: add top_level_section_title_classifier/test_selected_filings.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Elijas committed Dec 16, 2023
1 parent 6a44185 commit aaa7e65
Showing 1 changed file with 21 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
from pathlib import Path

import yaml

from tests.utils import traverse_repository_for_filings

DEFAULT_YAML = Path(__file__).parent / "selected-filings.yaml"


def test_filings_exist():
with DEFAULT_YAML.open("r") as file:
selected_filings = yaml.safe_load(file)
accession_numbers = list(selected_filings["accession_numbers"])
assert accession_numbers, "No accession numbers found in YAML file."

existing_numbers = {
report.accession_number for report in traverse_repository_for_filings()
}

for accession_number in accession_numbers:
assert accession_number in existing_numbers, f"Missing {accession_number}"

0 comments on commit aaa7e65

Please sign in to comment.