From 5d7c5d20dd7fdd1ad81c6e328510767b1f33da4b Mon Sep 17 00:00:00 2001 From: Sebastian Muszynski Date: Sat, 27 Jan 2018 18:06:50 +0100 Subject: [PATCH] Empty state added and dummy implementation used. --- miio/tests/test_chuangmi_ir.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/miio/tests/test_chuangmi_ir.py b/miio/tests/test_chuangmi_ir.py index 5890f56a8..99fc7ed2b 100644 --- a/miio/tests/test_chuangmi_ir.py +++ b/miio/tests/test_chuangmi_ir.py @@ -11,20 +11,13 @@ class DummyChuangmiIr(ChuangmiIr): def __init__(self, *args, **kwargs): + self.state = {} self.return_values = { 'miIO.ir_learn': lambda x: True, 'miIO.ir_read': lambda x: True, 'miIO.ir_play': lambda x: self._ir_play_input_validation, } - self.start_state = self.state.copy() - - def send(self, command: str, parameters=None, retry_count=3): - """Overridden send() to return values from `self.return_values`.""" - return self.return_values[command](parameters) - - def _reset_state(self): - """Revert back to the original state.""" - self.state = self.start_state.copy() + super().__init__(args, kwargs) @staticmethod def _ir_play_input_validation(self, props):