From 2c0478b0de925c56f275aee4f89abb41f81025c4 Mon Sep 17 00:00:00 2001 From: Jason Rumney Date: Sat, 19 Jun 2021 20:46:43 +1200 Subject: [PATCH] Update tests for change of state from datetime to string. --- tests/test_sensor.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/test_sensor.py b/tests/test_sensor.py index 3a755d8..2d57b0b 100644 --- a/tests/test_sensor.py +++ b/tests/test_sensor.py @@ -143,8 +143,7 @@ async def test_async_update_success(hass, aioclient_mock): assert sensor.icon == "mdi:train" assert sensor.name == "Metlink WELL" assert sensor.unique_id == "metlink_WELL_rKPL_dPorirua" - time = dt_util.parse_datetime(expected["departure"]) - assert sensor.state == time + assert sensor.state == expected["departure"] async def test_async_update_failed(): @@ -219,8 +218,7 @@ async def test_async_update_multiple(hass, aioclient_mock): assert sensor.icon == "mdi:train" assert sensor.name == "Metlink WELL" assert sensor.unique_id == "metlink_WELL_rKPL" - time = dt_util.parse_datetime(expected["departure"]) - assert sensor.state == time + assert sensor.state == expected["departure"] def test_slug():