def METHOD_NAME(self):
    """Test that avg level 0 measurement data is correctly marginalized."""
    memory = np.asarray(
        [[-1059254.375, -26266612.0], [-9012669.0, -41877468.0], [6027076.0, -54875060.0]],
        dtype=complex,
    )
    result = marginal_memory(memory, [0, 2], avg_data=True)
    expected = np.asarray(
        [[-1059254.375, -26266612.0], [6027076.0, -54875060.0]], dtype=complex
    )
    np.testing.assert_array_equal(result, expected)