def METHOD_NAME():
    x = cirq.XPowGate(dimension=3)
    sim = CustomStateSimulator(ComputationalBasisSimState, noise=x**2)
    circuit = create_test_circuit()
    r = sim.simulate(circuit)
    assert r.measurements == {'a': np.array([2]), 'b': np.array([2])}
    assert r._final_simulator_state._state.basis == [1, 2]