def METHOD_NAME(self):
    """Tests that create_initial_state works with the StatePrep operator."""
    prep_op = qml.StatePrep(np.array([0, 1, 0, 0, 0, 0, 0, 1]) / np.sqrt(2), wires=[0, 1, 2])
    state = create_initial_state([0, 1, 2], prep_operation=prep_op)
    assert state[0, 0, 1] == 1 / np.sqrt(2)
    assert state[1, 1, 1] == 1 / np.sqrt(2)
    state[0, 0, 1] = 0
    state[1, 1, 1] = 0  # set to zero to make test below simple
    assert qml.math.allequal(state, np.zeros((2, 2, 2)))