b197fb0946afe59b45a8c7a20a567ebfe253108f,neurodsp/tests/test_utils.py,,test_demean,#,11

Before Change

    dat = demean(dat)

    // TODO: Add isclose checking here
    assert True

def test_normalize_variance():

    dat = np.array([2, 2, 2])

After Change

def test_demean():

    d1 = np.array([1, 2, 3])
    d2 = np.array([0, 1, 2, 3, 0])

    // Check default - demean to 0
    out1 = demean(d1)
    assert np.isclose(out1.mean(), 0.)

    // Check demeaning and adding specific mean
    out2 = demean(d1, mean=1.)
    assert np.isclose(out2.mean(), 1.)

    // Check dealing with zero entries
    out3 = demean(d2)
    assert np.isclose(out3[np.nonzero(out3)].mean(), 0)

    // Check turning of non-zero selection
    out3 = demean(d2, mean=1, select_nonzero = False)

Instances