Python Mock.key方法代码示例

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
def test_gamestate_handle_events_pause_quit(Dungeon, event_get, fontinit):
    waves = MagicMock()
    waves.__len__.return_value = 2

    def event_get_called():
        return cur_events

    event_get.side_effect = event_get_called

    d = Mock()
    d.walls = []
    Dungeon.load.return_value = d

    gs = gamestate.GameState()
    assert not gs.paused
    assert gs.running

    # No events, no changes.
    cur_events = []
    gs.handle_events()
    assert not gs.paused
    assert gs.running
    assert event_get.call_count == 1

    # 'quit' -- q -> not-running
    event_key_q = Mock()
    event_key_q.type = pygame.KEYDOWN
    event_key_q.key = pygame.K_q
    cur_events = [event_key_q]
    gs.handle_events()
    assert not gs.paused
    assert not gs.running
    assert event_get.call_count == 2

    # 'pause' -- p -> paused, p -> unpaused
    gs = gamestate.GameState()
    event_key_p = Mock()
    event_key_p.type = pygame.KEYDOWN
    event_key_p.key = pygame.K_p
    cur_events = [event_key_p]
    gs.handle_events()
    assert gs.paused
    assert gs.running

    # (toggle-unpaused)
    gs.handle_events()
    assert not gs.paused
    assert gs.running

    # (toggle-paused, then quit while paused)
    gs.handle_events()
    assert gs.paused
    assert gs.running
    cur_events = [event_key_q]
    gs.handle_events()
    assert gs.paused
    assert not gs.running

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_fire(self, mock_load_png_sequence, mock_pulsator,
                  mock_bullet_class):
        img, rect = Mock(), Mock()
        mock_image_sequence = [(img, rect)]
        mock_load_png_sequence.return_value = mock_image_sequence
        mock_paddle = Mock()
        mock_paddle.rect.center = (100, 100)
        mock_paddle.rect.bottomleft = (60, 120)
        mock_paddle.rect.width = 50
        mock_game = Mock()
        mock_event = Mock()
        mock_event.key = pygame.K_SPACE
        mock_bullet1, mock_bullet2 = Mock(), Mock()
        mock_bullet_class.side_effect = mock_bullet1, mock_bullet2

        state = LaserState(mock_paddle, mock_game)
        state._fire(mock_event)

        mock_bullet_class.assert_has_calls(
            [call(mock_game, position=(70, 120)),
             call(mock_game, position=(100, 120))])
        self.assertEqual(len(state._bullets), 2)
        mock_game.sprites.append.assert_has_calls([call(mock_bullet1),
                                                   call(mock_bullet2)])
        mock_bullet1.release.assert_called_once_with()
        mock_bullet2.release.assert_called_once_with()

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
 def _get_recursive_structure(self, structure, path, parent):
     if not isinstance(structure, dict) or len(structure) == 0:
         parent.value = structure
         return
     for key, subdict in structure.items():
             child = Mock()
             child.value = None
             child.key = path + "/" + key
             parent.children.append(child)
             self._get_recursive_structure(subdict, child.key, child)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_release_ball(self, mock_pygame, mock_load_png_sequence):
        mock_game = _configure_mocks(mock_load_png_sequence, mock_pygame)
        mock_game.balls = [Mock(), Mock(), Mock()]
        mock_event = Mock()
        mock_event.key = mock_pygame.K_SPACE  # Spacebar event

        catch_powerup = CatchPowerUp(mock_game, Mock())
        catch_powerup._release_ball(mock_event)

        for ball in mock_game.balls:
            ball.release.assert_called_once_with()

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_release_ball_when_other_key_pressed(self, mock_pygame,
                                                 mock_load_png_sequence):
        mock_game = _configure_mocks(mock_load_png_sequence, mock_pygame)
        mock_game.balls = [Mock(), Mock(), Mock()]
        mock_event = Mock()
        mock_event.key = mock_pygame.SOME_OTHER_KEY

        catch_powerup = CatchPowerUp(mock_game, Mock())
        catch_powerup._release_ball(mock_event)

        for ball in mock_game.balls:
            self.assertEqual(ball.call_count, 0)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
def do_event(tool, etype, button=1, xdata=0, ydata=0, key=None, step=1):
    """
     *name*
        the event name

    *canvas*
        the FigureCanvas instance generating the event

    *guiEvent*
        the GUI event that triggered the matplotlib event

    *x*
        x position - pixels from left of canvas

    *y*
        y position - pixels from bottom of canvas

    *inaxes*
        the :class:`~matplotlib.axes.Axes` instance if mouse is over axes

    *xdata*
        x coord of mouse in data coords

    *ydata*
        y coord of mouse in data coords

     *button*
        button pressed None, 1, 2, 3, 'up', 'down' (up and down are used
        for scroll events)

    *key*
        the key depressed when the mouse event triggered (see
        :class:`KeyEvent`)

    *step*
        number of scroll steps (positive for 'up', negative for 'down')
    """
    event = Mock()
    event.button = button
    ax = tool.ax
    event.x, event.y = ax.transData.transform([(xdata, ydata),
                                               (xdata, ydata)])[00]
    event.xdata, event.ydata = xdata, ydata
    event.inaxes = ax
    event.canvas = ax.figure.canvas
    event.key = key
    event.step = step
    event.guiEvent = None
    event.name = 'Custom'

    func = getattr(tool, etype)
    func(event)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_operations_evaluated(self, mock_fetch_data: Mock, *mocks):
        mock_operation = Mock(name='mock_operation ', spec=f.Operation)
        mock_operation.key, mock_operation.definition = 'mock_operation', slicer.table.abc
        mock_operation.metrics = []

        mock_widget = f.Widget(mock_operation)
        mock_widget.transform = Mock()

        mock_df = {}
        mock_fetch_data.return_value = mock_df

        # Need to keep widget the last call in the chain otherwise the object gets cloned and the assertion won't work
        slicer.data \
            .dimension(slicer.dimensions.timestamp) \
            .widget(mock_widget) \
            .fetch()

        mock_operation.apply.assert_called_once_with(mock_df, None)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_fire_no_space(self, mock_load_png_sequence, mock_pulsator,
                           mock_bullet_class):
        """Test that fire does not happen when spacebar not pressed.
        """
        img, rect = Mock(), Mock()
        mock_image_sequence = [(img, rect)]
        mock_load_png_sequence.return_value = mock_image_sequence
        mock_paddle = Mock()
        mock_paddle.rect.center = (100, 100)
        mock_paddle.rect.bottomleft = (60, 120)
        mock_paddle.rect.width = 50
        mock_game = Mock()
        mock_event = Mock()
        mock_event.key = pygame.KEYUP

        state = LaserState(mock_paddle, mock_game)
        state._fire(mock_event)

        self.assertEqual(mock_bullet_class.call_count, 0)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_operations_results_stored_in_data_frame(self, mock_fetch_data: Mock, *mocks):
        mock_operation = Mock(name='mock_operation ', spec=f.Operation)
        mock_operation.key, mock_operation.definition = 'mock_operation', slicer.table.abc
        mock_operation.metrics = []

        mock_widget = f.Widget(mock_operation)
        mock_widget.transform = Mock()

        mock_df = {}
        mock_fetch_data.return_value = mock_df

        # Need to keep widget the last call in the chain otherwise the object gets cloned and the assertion won't work
        slicer.data \
            .dimension(slicer.dimensions.timestamp) \
            .widget(mock_widget) \
            .fetch()

        f_op_key = format_metric_key(mock_operation.key)
        self.assertIn(f_op_key, mock_df)
        self.assertEqual(mock_df[f_op_key], mock_operation.apply.return_value)

# 需要导入模块: from unittest.mock import Mock [as 别名]
# 或者: from unittest.mock.Mock import key [as 别名]
    def test_fire_max(self, mock_load_png_sequence, mock_pulsator,
                      mock_bullet_class):
        """Test that it is not possible to fire 2 more bullets if there are
        already 3 or more in the air.
        """
        img, rect = Mock(), Mock()
        mock_image_sequence = [(img, rect)]
        mock_load_png_sequence.return_value = mock_image_sequence
        mock_paddle = Mock()
        mock_paddle.rect.center = (100, 100)
        mock_paddle.rect.bottomleft = (60, 120)
        mock_paddle.rect.width = 50
        mock_game = Mock()
        mock_event = Mock()
        mock_event.key = pygame.K_SPACE

        state = LaserState(mock_paddle, mock_game)
        state._bullets.extend([Mock()] * 3)
        state._fire(mock_event)

        self.assertEqual(mock_bullet_class.call_count, 0)