diff --git a/app/main/views/notifications.py b/app/main/views/notifications.py index 0f2ae3b5e..873ff4397 100644 --- a/app/main/views/notifications.py +++ b/app/main/views/notifications.py @@ -1,8 +1,8 @@ # -*- coding: utf-8 -*- import base64 +import os from datetime import datetime -import os from flask import ( Response, abort, @@ -13,8 +13,8 @@ from flask import ( url_for, ) from flask_login import login_required - from notifications_python_client.errors import APIError + from app import ( current_service, format_date_numeric, diff --git a/tests/app/main/views/test_notifications.py b/tests/app/main/views/test_notifications.py index c82702b7c..0f89d2de4 100644 --- a/tests/app/main/views/test_notifications.py +++ b/tests/app/main/views/test_notifications.py @@ -1,14 +1,16 @@ import base64 +from functools import partial +from unittest.mock import mock_open import pytest from flask import url_for from freezegun import freeze_time -from functools import partial -import pytest -from unittest.mock import mock_open - from notifications_python_client.errors import APIError -from tests.conftest import mock_get_notification, SERVICE_ONE_ID, normalize_spaces +from tests.conftest import ( + SERVICE_ONE_ID, + mock_get_notification, + normalize_spaces, +) @pytest.mark.parametrize('notification_status, expected_status', [