def METHOD_NAME(self):
    """Test that avg memory level 1 data is correctly marginalized."""
    memory = np.array([1.0j, 1.0, 0.5 + 0.5j], dtype=complex)
    result = marginal_memory(memory, [0, 1])
    expected = np.array([1.0j, 1.0], dtype=complex)
    np.testing.assert_array_equal(result, expected)