コード例 #1
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
    def test_resize_shrink(self):
        d = Dist(5)
        for i in range(len(d)):
            d[i] = i + 1
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())

        d.resize(3)
        self.assertEqual(3, len(d))
        self.assertEqual(6, d.counts())
        for i in range(len(d)):
            self.assertEqual(i + 1, d[i])
コード例 #2
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
    def test_resize_shrink(self):
        d = Dist(5)
        for i in range(len(d)):
            d[i] = i+1
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())

        d.resize(3)
        self.assertEqual(3, len(d))
        self.assertEqual(6, d.counts())
        for i in range(len(d)):
            self.assertEqual(i+1, d[i])
コード例 #3
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
    def test_resize_grow(self):
        d = Dist(3)
        for i in range(len(d)):
            d[i] = i + 1
        self.assertEqual(3, len(d))
        self.assertEqual(6, d.counts())

        d.resize(5)
        self.assertEqual(5, len(d))
        self.assertEqual(6, d.counts())
        for i in range(3):
            self.assertEqual(i + 1, d[i])
        for i in range(3, len(d)):
            self.assertEqual(0, d[i])
コード例 #4
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
    def test_resize_grow(self):
        d = Dist(3)
        for i in range(len(d)):
            d[i] = i+1
        self.assertEqual(3, len(d))
        self.assertEqual(6, d.counts())

        d.resize(5)
        self.assertEqual(5, len(d))
        self.assertEqual(6, d.counts())
        for i in range(3):
            self.assertEqual(i+1, d[i])
        for i in range(3,len(d)):
            self.assertEqual(0, d[i])
コード例 #5
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
    def test_copy(self):
        d = Dist(5)
        for i in range(len(d)):
            d[i] = i + 1
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())

        e = d.copy()
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())
        for i in range(len(d)):
            self.assertEqual(e[i], d[i])

        d[0] = 5
        self.assertNotEqual(e[0], d[0])
コード例 #6
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
    def test_copy(self):
        d = Dist(5)
        for i in range(len(d)):
            d[i] = i+1
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())

        e = d.copy()
        self.assertEqual(5, len(d))
        self.assertEqual(15, d.counts())
        for i in range(len(d)):
            self.assertEqual(e[i], d[i])

        d[0] = 5
        self.assertNotEqual(e[0], d[0])
コード例 #7
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
    def test_tick(self):
        d = Dist(2)

        self.assertEqual(1, d.tick(0))
        self.assertEqual(2, d.tick(0))
        self.assertEqual(2, d.counts())
        self.assertTrue(d.valid())
コード例 #8
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
    def test_counts(self):
        d = Dist(2)
        self.assertEqual(0, d.counts())

        d[0] = 3
        self.assertEqual(3, d.counts())

        d[0] = 2
        self.assertEqual(2, d.counts())

        d[1] = 3
        self.assertEqual(5, d.counts())

        d[0] = 0
        d[1] = 0
        self.assertEqual(0, d.counts())
コード例 #9
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
 def test_dump(self):
     d = Dist(5)
     for i in range(1, len(d)):
         d[i] = i+1
     self.assertEqual(14, d.counts())
     probs = d.dump()
     self.assertTrue((probs == np.array([0., 2./14, 3./14, 4./14, 5./14])).all())
コード例 #10
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
    def test_tick(self):
        d = Dist(2)

        self.assertEqual(1, d.tick(0))
        self.assertEqual(2, d.tick(0))
        self.assertEqual(2, d.counts())
        self.assertTrue(d.valid())
コード例 #11
0
ファイル: test_dist.py プロジェクト: jakehanson/PyInform
    def test_counts(self):
        d = Dist(2)
        self.assertEqual(0, d.counts())

        d[0] = 3
        self.assertEqual(3, d.counts())

        d[0] = 2
        self.assertEqual(2, d.counts())

        d[1] = 3
        self.assertEqual(5, d.counts())

        d[0] = 0
        d[1] = 0
        self.assertEqual(0, d.counts())
コード例 #12
0
ファイル: test_dist.py プロジェクト: sehernandezd/PyInform
 def test_dump(self):
     d = Dist(5)
     for i in range(1, len(d)):
         d[i] = i + 1
     self.assertEqual(14, d.counts())
     probs = d.dump()
     self.assertTrue(
         (probs == np.array([0., 2. / 14, 3. / 14, 4. / 14, 5. / 14])).all())