from pathlib import Path
import sys


SCRIPTS = Path(__file__).resolve().parents[1] / "scripts"
sys.path.insert(0, str(SCRIPTS))

from source_classification import classify_npdes_facility  # noqa: E402


def test_current_violation_is_not_labeled_as_proven_pollution_source():
    classification = classify_npdes_facility(current_violation="Yes")

    assert classification.evidence_state == "compliance_flag_current"
    assert classification.public_label == "NPDES facility with current compliance flag"
    assert "polluter" not in classification.public_label.casefold()