def setUp(self):
        # Construct with mocked _send_restyle_msg method
        self.figure = go.Figure(
            data=[
                go.Scatter(),
                go.Bar(),
                go.Parcoords(dimensions=[{}, {"label": "dim 2"}, {}]),
            ]
        )

        # Mock out the message method
        self.figure._send_restyle_msg = MagicMock()