def test_more_bins(): np.random.seed(42) assert_array_almost_equal( even_width_lags(np.random.normal(5, 1, 1000), 10, None), np.array([0.88527315, 1.7705463, 2.65581945, 3.5410926, 4.42636575, 5.3116388, 6.19691204, 7.08218519, 7.96745834, 8.8527314]) )
def test_median_split(): np.random.seed(42) bins, _ = even_width_lags(np.random.normal(5, 1, 1000), 2, None) assert_array_almost_equal(bins, np.array([4.42636575, 8.85273149]))
def test_too_large_maxlag(): np.random.seed(42) bins, n = even_width_lags(np.random.normal(5, 1, 1000), 4, 400) assert_array_almost_equal( bins, np.array([2.21318287, 4.42636575, 6.63954862, 8.85273149]))
def test_maxlag(): np.random.seed(42) bins, _ = even_width_lags(np.random.normal(5, 1, 1000), 4, 4.4) assert_array_almost_equal(bins, np.array([1.1, 2.2, 3.3, 4.4]))
def test_normal(): np.random.seed(42) assert_array_almost_equal( even_width_lags(np.random.normal(5, 1, 1000), 4, None), np.array([2.21318287, 4.42636575, 6.63954862, 8.85273149]) )