コード例 #1
0
ファイル: tests.py プロジェクト: craigds/pyGVM
    def test_first_point(self):
        cs = Clusters(dimension=2, capacity=2)

        cs.add(1, (5, 5), 'bert')
        self.assertEqual(len(cs.clusters), 1)
        self.assertEqual(len(cs.clusters[0]), 1)

        self.assertEqual(cs.clusters[0].mass, 1)
        self.assertEqual(cs.clusters[0].center, (5, 5))
コード例 #2
0
ファイル: tests.py プロジェクト: craigds/pyGVM
    def test_same_point_multiple_times(self):
        # NOTE: this is stupid. If you seed both clusters with the same
        # point, your clustering will basically just be random.
        cs = Clusters(dimension=2, capacity=2)

        cs.add(1, (5, 5), 'tom')
        self.assertEqual(len(cs.clusters), 1)
        self.assertEqual(len(cs.clusters[0]), 1)

        cs.add(1, (5, 5), 'dick')
        self.assertEqual(len(cs.clusters), 2)
        self.assertEqual(len(cs.clusters[0]), 1)
        self.assertEqual(len(cs.clusters[1]), 1)

        cs.add(1, (5, 5), 'harry')
        self.assertEqual(len(cs.clusters), 2)
        self.assertEqual(len(cs.clusters[0]), 2)
        self.assertEqual(len(cs.clusters[1]), 1)

        self.assertEqual(cs.clusters[0].mass, 2)
        self.assertEqual(cs.clusters[1].mass, 1)