Add __init__ files to test packages

This ensures that mypy will typecheck them. mypy doesn't recursively
check directories without an __init__ file. See: python/mypy#6385

tests/routes_tests/test_audit_status.py

@@ -1,6 +1,7 @@
 import json
 import pytest
 from flask.testing import FlaskClient
+from typing import Generator
 
 from tests.helpers import (
     compare_json,
@@ -14,7 +15,7 @@
 
 
 @pytest.fixture()
-def election_id(client: FlaskClient) -> str:
+def election_id(client: FlaskClient) -> Generator[str, None, None]:
     yield create_election(client, is_multi_jurisdiction=False)
 
 

tests/routes_tests/test_auth.py

@@ -130,7 +130,7 @@ def test_auditadmin_callback(
     rv = client.get("/auth/auditadmin/callback?code=foobar")
     assert rv.status_code == 302
 
-    with client.session_transaction() as session:
+    with client.session_transaction() as session:  # type: ignore
         assert session["_user"]["type"] == UserType.AUDIT_ADMIN
         assert session["_user"]["email"] == AA_EMAIL
 
@@ -155,7 +155,7 @@ def test_jurisdictionadmin_callback(
     rv = client.get("/auth/jurisdictionadmin/callback?code=foobar")
     assert rv.status_code == 302
 
-    with client.session_transaction() as session:
+    with client.session_transaction() as session:  # type: ignore
         assert session["_user"]["type"] == UserType.JURISDICTION_ADMIN
         assert session["_user"]["email"] == JA_EMAIL
 
@@ -168,7 +168,7 @@ def test_logout(client: FlaskClient):
 
     rv = client.get("/auth/logout")
 
-    with client.session_transaction() as session:
+    with client.session_transaction() as session:  # type: ignore
         assert session["_user"] is None
 
     assert rv.status_code == 302

tests/routes_tests/test_ballot_list.py

@@ -1,6 +1,6 @@
 import json
 from flask.testing import FlaskClient
-
+from typing import Generator
 import pytest
 
 from tests.helpers import assert_ok, post_json, create_election
@@ -9,7 +9,7 @@
 
 
 @pytest.fixture()
-def election_id(client: FlaskClient) -> str:
+def election_id(client: FlaskClient) -> Generator[str, None, None]:
     yield create_election(client, is_multi_jurisdiction=False)
 
 

tests/routes_tests/test_contests.py

@@ -183,7 +183,7 @@ def test_contests_round_status(
 def test_contests_missing_field(
     client: FlaskClient, election_id: str, jurisdiction_ids: List[str]
 ):
-    contest = {
+    contest: JSONDict = {
         "id": str(uuid.uuid4()),
         "name": "Contest 1",
         "isTargeted": True,

tests/routes_tests/test_jurisdictions.py

@@ -4,7 +4,7 @@
 
 import json, io, uuid
 from datetime import datetime
-from typing import List
+from typing import List, Generator
 
 from tests.helpers import (
     assert_ok,

tests/routes_tests/test_report.py

@@ -1,6 +1,6 @@
 import json, random
 from flask.testing import FlaskClient
-
+from typing import Generator
 import pytest
 
 from tests.helpers import assert_ok, post_json, create_election
@@ -9,7 +9,7 @@
 
 
 @pytest.fixture()
-def election_id(client: FlaskClient) -> str:
+def election_id(client: FlaskClient) -> Generator[str, None, None]:
     yield create_election(client, is_multi_jurisdiction=False)
 
 

tests/util_tests/test_binpacking.py

@@ -72,7 +72,7 @@ def test_init(self, bucket):
         )
         assert (
             not bucket.largest_element
-        ), "Initial largest element was not None".format(bucket.largest_element)
+        ), "Initial largest element was not None: {}".format(bucket.largest_element)
 
     def test_add_batch(self, bucket):
         expected_batches = {"1": 100}