Fix tests

rytilahti · Aug 26, 2021 · 108f7ba · 108f7ba
1 parent 7026cc7
commit 108f7ba
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/miio/tests/test_vacuum.py b/miio/tests/test_vacuum.py
@@ -233,9 +233,9 @@ def test_history_details(self):
             "send",
             return_value=[[1488347071, 1488347123, 16, 0, 0, 0]],
         ):
-            assert self.device.clean_details(
-                123123, return_list=False
-            ).duration == datetime.timedelta(seconds=16)
+            assert self.device.clean_details(123123).duration == datetime.timedelta(
+                seconds=16
+            )
 
     def test_history_details_dict(self):
         with patch.object(
@@ -256,9 +256,9 @@ def test_history_details_dict(self):
                 }
             ],
         ):
-            assert self.device.clean_details(
-                123123, return_list=False
-            ).duration == datetime.timedelta(seconds=950)
+            assert self.device.clean_details(123123).duration == datetime.timedelta(
+                seconds=950
+            )
 
     def test_history_empty(self):
         with patch.object(