コード例 #1
0
ファイル: test_random.py プロジェクト: duhaime/lightning
def test_randint():
    rs = RandomState(seed=0)
    vals = [rs.randint(10) for t in xrange(10000)]
    assert_almost_equal(np.mean(vals), 5.018)
コード例 #2
0
ファイル: test_random.py プロジェクト: duhaime/lightning
def test_shuffle():
    ind = np.arange(10)
    rs = RandomState(seed=0)
    rs.shuffle(ind)
    assert_array_equal(ind, [2, 8, 4, 9, 1, 6, 7, 3, 0, 5])