コード例 #1
0
ファイル: test_sampler.py プロジェクト: ainafp/pyhrf
    def test_burnin(self):

        v = np.array([1, 2])
        nb_its = 10
        pace = 1
        start = 5
        t = Trajectory(v, ['time'], {'time':np.array([1,2])},
                       pace, start, nb_its, first_saved_iteration=start)

        for i in xrange(nb_its):
            v *= 2
            t.record(i)

        assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its)
        self.assertEqual(t.saved_iterations, range(start,nb_its))
コード例 #2
0
ファイル: test_sampler.py プロジェクト: Solvi/pyhrf
    def test_basic(self):

        v = np.array([1, 2])
        nb_its = 10
        pace = 1
        start = 0
        t = Trajectory(v, ['time'], {'time':np.array([1,2])},
                       pace, start, nb_its)

        for i in xrange(nb_its):
            v *= 2
            t.increment(i)

        assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its)
        self.assertEqual(t.saved_iterations, [-1] + range(start,nb_its))
コード例 #3
0
ファイル: test_sampler.py プロジェクト: zddzxxsmile/pyhrf
    def test_basic(self):

        v = np.array([1, 2])
        nb_its = 10
        pace = 1
        start = 0
        t = Trajectory(v, ['time'], {'time': np.array([1, 2])}, pace, start,
                       nb_its)

        for i in xrange(nb_its):
            v *= 2
            t.record(i)

        assert_array_equal(t.get_last(), np.array([1, 2]) * 2**nb_its)
        self.assertEqual(t.saved_iterations, [-1] + range(start, nb_its))