Initial build out

tests/test_irs990_titles.py

@@ -0,0 +1,45 @@
+from admin_analytics.irs990.titles import normalize_title
+
+
+def test_president():
+    assert normalize_title("PRESIDENT") == "PRESIDENT"
+    assert normalize_title("President & CEO") == "PRESIDENT"
+
+
+def test_vice_president_finance():
+    assert normalize_title("VICE PRESIDENT FOR FINANCE") == "VP_FINANCE"
+    assert normalize_title("VP Finance and Administration") == "VP_FINANCE"
+    assert normalize_title("EVP & VP for Business") == "VP_FINANCE"
+
+
+def test_vice_president_other():
+    assert normalize_title("VICE PRESIDENT") == "VP_OTHER"
+
+
+def test_provost():
+    assert normalize_title("PROVOST") == "PROVOST"
+    assert normalize_title("Executive Vice President and Provost") == "PROVOST"
+
+
+def test_trustee():
+    assert normalize_title("TRUSTEE") == "TRUSTEE"
+
+
+def test_dean():
+    assert normalize_title("DEAN OF ENGINEERING") == "DEAN"
+
+
+def test_cfo():
+    assert normalize_title("CHIEF FINANCIAL OFFICER") == "CFO"
+    assert normalize_title("CFO") == "CFO"
+
+
+def test_other():
+    assert normalize_title("ATHLETIC DIRECTOR") == "DIRECTOR"
+    assert normalize_title("FOOTBALL COACH") == "OTHER"
+
+
+def test_none_and_empty():
+    assert normalize_title(None) is None
+    assert normalize_title("") is None
+    assert normalize_title("  ") is None