def METHOD_NAME(self, calc_psib):
    mypcs = self.MyPulsedCoilset(self.params, *[None] * 4)
    def r_premag(self, val=1):
        self.snapshots[self.BREAKDOWN] = mock.MagicMock()
        self.snapshots[self.BREAKDOWN].eq.breakdown_psi = val
    with mock.patch.object(
        self.MyPulsedCoilset, "run_premagnetisation", new=r_premag
    ):
        # all the same because calc_psib is mocked
        out_val = mypcs.calculate_sof_eof_fluxes()
        np.testing.assert_allclose(out_val, (0, -63))
        r_premag(mypcs, 2)
        mypcs.calculate_sof_eof_fluxes()
        np.testing.assert_allclose(out_val, (0, -63))
        mypcs.calculate_sof_eof_fluxes(3)
        np.testing.assert_allclose(out_val, (0, -63))
    # calc_psib called with different values
    psi_premag2pi = np.array([call[0][0] for call in calc_psib.call_args_list])
    np.testing.assert_allclose(psi_premag2pi[0] * 2, psi_premag2pi[1])
    np.testing.assert_allclose(psi_premag2pi[0] * 3, psi_premag2pi[2])