summaryrefslogtreecommitdiff
path: root/src/cython/test/test_subsampling.py
blob: 2dc12a891a6b18fb8c34a6c97e83f44011c285c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import gudhi

"""This file is part of the Gudhi Library. The Gudhi library
   (Geometric Understanding in Higher Dimensions) is a generic C++
   library for computational topology.

   Author(s):       Vincent Rouvreau

   Copyright (C) 2016 INRIA

   This program is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

__author__ = "Vincent Rouvreau"
__copyright__ = "Copyright (C) 2016 INRIA"
__license__ = "GPL v3"


def test_write_off_file_for_tests():
    file = open("subsample.off", "w")
    file.write("nOFF\n")
    file.write("2 7 0 0\n")
    file.write("1.0 1.0\n")
    file.write("7.0 0.0\n")
    file.write("4.0 6.0\n")
    file.write("9.0 6.0\n")
    file.write("0.0 14.0\n")
    file.write("2.0 19.0\n")
    file.write("9.0 17.0\n")
    file.close()

def test_simple_choose_n_farthest_points_with_a_starting_point():
    point_set = [[0,1], [0,0], [1,0], [1,1]]
    i = 0
    for point in point_set:
        # The iteration starts with the given starting point
        sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = 1, starting_point = i)
        assert sub_set[0] == point_set[i]
        i = i + 1

    # The iteration finds then the farthest
    sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = 2, starting_point = 1)
    assert sub_set[1] == point_set[3]
    sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = 2, starting_point = 3)
    assert sub_set[1] == point_set[1]
    sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = 2, starting_point = 0)
    assert sub_set[1] == point_set[2]
    sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = 2, starting_point = 2)
    assert sub_set[1] == point_set[0]

    # Test the limits
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 0, starting_point = 0) == []
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 1, starting_point = 0) == []
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 0, starting_point = 1) == []
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 1, starting_point = 1) == []

    # From off file test
    for i in range (0, 7):
        assert len(gudhi.choose_n_farthest_points(off_file = 'subsample.off', nb_points = i, starting_point = i)) == i

def test_simple_choose_n_farthest_points_randomed():
    point_set = [[0,1], [0,0], [1,0], [1,1]]
    # Test the limits
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 0) == []
    assert gudhi.choose_n_farthest_points(points = [], nb_points = 1) == []
    assert gudhi.choose_n_farthest_points(points = point_set, nb_points = 0) == []

    # Go furter than point set on purpose
    for iter in range(1,10):
        sub_set = gudhi.choose_n_farthest_points(points = point_set, nb_points = iter)
        for sub in sub_set:
            found = False
            for point in point_set:
                if point == sub:
                    found = True
            # Check each sub set point is existing in the point set
            assert found == True

    # From off file test
    for i in range (0, 7):
        assert len(gudhi.choose_n_farthest_points(off_file = 'subsample.off', nb_points = i)) == i

def test_simple_pick_n_random_points():
    point_set = [[0,1], [0,0], [1,0], [1,1]]
    # Test the limits
    assert gudhi.pick_n_random_points(points = [], nb_points = 0) == []
    assert gudhi.pick_n_random_points(points = [], nb_points = 1) == []
    assert gudhi.pick_n_random_points(points = point_set, nb_points = 0) == []

    # Go furter than point set on purpose
    for iter in range(1,10):
        sub_set = gudhi.pick_n_random_points(points = point_set, nb_points = iter)
        print(5)
        for sub in sub_set:
            found = False
            for point in point_set:
                if point == sub:
                    found = True
            # Check each sub set point is existing in the point set
            assert found == True