def test_raw_input_edit_cmd(self, ignored): mocked_cmd = Mock() with patch.dict(self.pymp.commands, {'\e': mocked_cmd}): self.pymp.raw_input('>>> ') mocked_cmd.assert_called_once_with('code')