Fix linting errors

This commit is contained in:
Neil Martinsen-Burrell 2022-11-23 13:56:59 -06:00
parent fd687b8e3b
commit 45e1dcbba2
No known key found for this signature in database
GPG key ID: 6A3C818CC10D0184
2 changed files with 6 additions and 7 deletions

View file

@ -4,6 +4,8 @@ from __future__ import annotations # allows forward references in annotations
import logging
from typing import Union
from django import forms
from django.shortcuts import render
@ -72,7 +74,7 @@ class OrganizationElectionForm(RegistrarForm):
(False, "No"),
],
),
required=False
required=False,
)
@ -298,12 +300,12 @@ def _get_organization_type(wizard: ApplicationWizard) -> Union[str, None]:
return None
def show_organization_federal(wizard: ApplicationWizard) -> Bool:
def show_organization_federal(wizard: ApplicationWizard) -> bool:
"""Show this step if the answer to the first question was "federal"."""
return _get_organization_type(wizard) == "Federal"
def show_organization_election(wizard: ApplicationWizard) -> Bool:
def show_organization_election(wizard: ApplicationWizard) -> bool:
"""Show this step if the answer to the first question implies it.
This shows for answers that aren't "Federal" or "Interstate".

View file

@ -134,7 +134,6 @@ class FormTests(TestWithUser, WebTest):
self.assertEquals(type_result.status_code, 302)
self.assertEquals(type_result["Location"], "/register/organization_federal/")
# ---- FEDERAL BRANCH PAGE ----
# Follow the redirect to the next form page
self.app.set_cookie(settings.SESSION_COOKIE_NAME, session_id)
@ -147,9 +146,7 @@ class FormTests(TestWithUser, WebTest):
federal_result = federal_form.submit()
self.assertEquals(federal_result.status_code, 302)
self.assertEquals(
federal_result["Location"], "/register/organization_contact/"
)
self.assertEquals(federal_result["Location"], "/register/organization_contact/")
# ---- ORG CONTACT PAGE ----
# Follow the redirect to the next form page