Skip to content

Commit

Permalink
#297 Some tidy-up
Browse files Browse the repository at this point in the history
  • Loading branch information
TWilkin committed Jun 4, 2023
1 parent 9a2fe38 commit 2f5d833
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions controllers/node/tests/device/test_local_node.py
Original file line number Diff line number Diff line change
@@ -1,20 +1,19 @@
from asyncio import Future
from typing import Callable, Dict, List, Tuple, Union
from typing import Any, Callable, Dict, List, Tuple, Union
from unittest.mock import PropertyMock, call

import pytest
from powerpi_common.device import DeviceStatus
from powerpi_common_test.device import DeviceTestBaseNew
from powerpi_common_test.device.mixin import (InitialisableMixinTestBaseNew,
PollableMixinTestBaseNew)
from powerpi_common_test.mqtt.mqtt import mock_producer
from powerpi_common_test.sensor.mixin import BatteryMixinTestBaseNew
from pytest_mock import MockerFixture

from node_controller.device.local_node import LocalNodeDevice

SubjectBuilder = Callable[
[Union[Dict, None], Union[Dict, None]],
[Union[Dict[str, Any], None], Union[Dict[str, Any], None]],
LocalNodeDevice
]

Expand Down Expand Up @@ -326,7 +325,7 @@ def subject_builder(
):
# pylint: disable=too-many-arguments

def build(pijuice: Union[Dict, None], pwm_fan: Union[Dict, None]):
def build(pijuice: Union[Dict[str, Any], None], pwm_fan: Union[Dict[str, Any], None]):
return LocalNodeDevice(
powerpi_config,
powerpi_logger,
Expand Down

0 comments on commit 2f5d833

Please sign in to comment.