mapclassify/tests/test_greedy.py

15     colors = greedy(world)
16     assert len(colors) == len(world)
17     assert set(colors) == set([0, 1, 2, 3, 4])
18     assert colors.value_counts().to_list() == [36, 36, 35, 35, 35]
19     assert (colors.index == world.index).all()
24     colors = greedy(
27     assert len(colors) == len(world)
28     assert set(colors) == set([0, 1, 2, 3, 4])
29     assert colors.value_counts().to_list() == [36, 36, 35, 35, 35]
34     colors = greedy(world, strategy="balanced", balance="area", min_distance=pysal_geos)
35     assert len(colors) == len(world)
36     assert set(colors) == set([0, 1, 2, 3, 4])
37     assert colors.value_counts().to_list() == [55, 49, 39, 32, 2]
42     colors = greedy(
45     assert len(colors) == len(world)
46     assert set(colors) == set([0, 1, 2, 3, 4])
47     assert colors.value_counts().to_list() == [39, 36, 36, 34, 32]
52     colors = greedy(
55     assert len(colors) == len(world)
56     assert set(colors) == set([0, 1, 2, 3, 4])
57     assert colors.value_counts().to_list() == [38, 36, 35, 34, 34]
62     colors = greedy(world, strategy="largest_first", min_distance=pysal_geos)
63     assert len(colors) == len(world)
64     assert set(colors) == set([0, 1, 2, 3, 4])
65     assert colors.value_counts().to_list() == [64, 49, 42, 21, 1]
70     colors = greedy(world, strategy="random_sequential", min_distance=pysal_geos)
71     assert len(colors) == len(world)
80     colors = greedy(world, strategy="smallest_last", min_distance=pysal_geos)
81     assert len(colors) == len(world)
82     assert set(colors) == set([0, 1, 2, 3])
83     assert colors.value_counts().to_list() == [71, 52, 39, 15]
88     colors = greedy(world, strategy="independent_set", min_distance=pysal_geos)
89     assert len(colors) == len(world)
90     assert set(colors) == set([0, 1, 2, 3, 4])
91     assert colors.value_counts().to_list() == [91, 42, 26, 13, 5]
96     colors = greedy(world, strategy="connected_sequential_bfs", min_distance=pysal_geos)
97     assert len(colors) == len(world)
98     assert set(colors) == set([0, 1, 2, 3, 4])
99     assert colors.value_counts().to_list() == [77, 46, 34, 18, 2]
104     colors = greedy(world, strategy="connected_sequential_dfs", min_distance=pysal_geos)
105     assert len(colors) == len(world)
106     assert set(colors) == set([0, 1, 2, 3, 4])
107     assert colors.value_counts().to_list() == [75, 52, 34, 14, 2]
112     colors = greedy(world, strategy="connected_sequential", min_distance=pysal_geos)
113     assert len(colors) == len(world)
114     assert set(colors) == set([0, 1, 2, 3, 4])
115     assert colors.value_counts().to_list() == [77, 46, 34, 18, 2]
120     colors = greedy(world, strategy="saturation_largest_first", min_distance=pysal_geos)
121     assert len(colors) == len(world)
122     assert set(colors) == set([0, 1, 2, 3])
123     assert colors.value_counts().to_list() == [71, 47, 42, 17]
128     colors = greedy(world, strategy="DSATUR", min_distance=pysal_geos)
129     assert len(colors) == len(world)
130     assert set(colors) == set([0, 1, 2, 3])
131     assert colors.value_counts().to_list() == [71, 47, 42, 17]
140     colors = greedy(world, sw="rook")
141     assert len(colors) == len(world)
142     assert set(colors) == set([0, 1, 2, 3, 4])
143     assert colors.value_counts().to_list() == [36, 36, 35, 35, 35]
147     colors = greedy(world, sw=sw)
148     assert len(colors) == len(world)
149     assert set(colors) == set([0, 1, 2, 3, 4])
150     assert colors.value_counts().to_list() == [36, 36, 35, 35, 35]
157     colors = greedy(reindexed, min_distance=pysal_geos)
158     assert len(colors) == len(world)
159     assert set(colors) == set([0, 1, 2, 3, 4])
160     assert colors.value_counts().to_list() == [36, 36, 35, 35, 35]
165     colors = greedy(europe, min_distance=500000)
166     assert len(colors) == len(europe)
167     assert set(colors) == set(range(13))
168     assert colors.value_counts().to_list() == [3] * 13