flake8

Signed-off-by: F.N. Claessen <felix@seita.nl>
FlexMeasures · Sep 11, 2023 · c4608cf · c4608cf
1 parent 60c15fa
commit c4608cf
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 5 deletions.
diff --git a/flexmeasures/api/v3_0/tests/test_api_v3_0_users.py b/flexmeasures/api/v3_0/tests/test_api_v3_0_users.py
@@ -1,6 +1,5 @@
 from flask import url_for
 from flask_login import current_user, logout_user
-from flask_security.core import AnonymousUser
 import pytest
 
 from flexmeasures.data.services.users import find_user_by_email

diff --git a/flexmeasures/api/v3_0/tests/test_assets_api.py b/flexmeasures/api/v3_0/tests/test_assets_api.py
@@ -320,7 +320,7 @@ def test_post_an_asset_with_invalid_data(client, setup_api_test_data, requesting
     )
 
     assert (
-        GenericAsset.query.filter_by(account_id=prosumer.id).count()
+        GenericAsset.query.filter_by(account_id=requesting_user.account.id).count()
         == num_assets_before
     )
 

diff --git a/flexmeasures/api/v3_0/tests/test_sensor_schedules.py b/flexmeasures/api/v3_0/tests/test_sensor_schedules.py
@@ -6,7 +6,7 @@
 from rq.job import Job
 
 from flexmeasures.api.common.responses import unknown_schedule, unrecognized_event
-from flexmeasures.api.tests.utils import check_deprecation, get_auth_token
+from flexmeasures.api.tests.utils import check_deprecation
 from flexmeasures.api.v3_0.tests.utils import message_for_trigger_schedule
 from flexmeasures.data.models.data_sources import DataSource
 from flexmeasures.data.models.generic_assets import GenericAsset
@@ -116,7 +116,6 @@ def test_trigger_and_get_schedule_with_unknown_prices(
     message,
     requesting_user,
 ):
-    auth_token = None
     sensor = add_battery_assets["Test battery"].sensors[0]
 
     # trigger a schedule through the /sensors/<id>/schedules/trigger [POST] api endpoint

diff --git a/flexmeasures/api/v3_0/tests/test_sensors_api.py b/flexmeasures/api/v3_0/tests/test_sensors_api.py
@@ -6,7 +6,6 @@
 
 from flexmeasures.data.models.time_series import TimedBelief
 from flexmeasures import Sensor
-from flexmeasures.api.tests.utils import get_auth_token
 from flexmeasures.api.v3_0.tests.utils import get_sensor_post_data
 from flexmeasures.data.schemas.sensors import SensorSchema