Skip to content

Commit b3fc84b

Browse files
committed
Fixed all plane plotters
1 parent e3e98ab commit b3fc84b

1 file changed

Lines changed: 80 additions & 0 deletions

File tree

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,80 @@
1+
from autoarray.plotters import plotter_util
2+
3+
4+
class TestRadii:
5+
def test__radii_bin_size_from_minimum_and_maximum_radii__is_correct_values(self):
6+
radii_bin_size = plotter_util.radii_bin_size_from_minimum_and_maximum_radii_and_radii_points(
7+
minimum_radius=0.0, maximum_radius=1.0, radii_points=1
8+
)
9+
10+
assert radii_bin_size == 1.0
11+
12+
radii_bin_size = plotter_util.radii_bin_size_from_minimum_and_maximum_radii_and_radii_points(
13+
minimum_radius=0.0, maximum_radius=1.0, radii_points=2
14+
)
15+
16+
assert radii_bin_size == 0.5
17+
18+
radii_bin_size = plotter_util.radii_bin_size_from_minimum_and_maximum_radii_and_radii_points(
19+
minimum_radius=0.0, maximum_radius=2.0, radii_points=1
20+
)
21+
22+
assert radii_bin_size == 2.0
23+
24+
radii_bin_size = plotter_util.radii_bin_size_from_minimum_and_maximum_radii_and_radii_points(
25+
minimum_radius=1.0, maximum_radius=6.0, radii_points=5
26+
)
27+
28+
assert radii_bin_size == 1.0
29+
30+
radii_bin_size = plotter_util.radii_bin_size_from_minimum_and_maximum_radii_and_radii_points(
31+
minimum_radius=100.0, maximum_radius=200.0, radii_points=25
32+
)
33+
34+
assert radii_bin_size == 4.0
35+
36+
def test__quantity_radii_from_minimum_and_maximum_radii__is_correct_values(self):
37+
quantity_radii = plotter_util.quantity_radii_from_minimum_and_maximum_radii_and_radii_points(
38+
minimum_radius=0.0, maximum_radius=1.0, radii_points=1
39+
)
40+
41+
assert quantity_radii == [0.0, 1.0]
42+
43+
quantity_radii = plotter_util.quantity_radii_from_minimum_and_maximum_radii_and_radii_points(
44+
minimum_radius=0.0, maximum_radius=1.0, radii_points=2
45+
)
46+
47+
assert quantity_radii == [0.0, 0.5, 1.0]
48+
49+
quantity_radii = plotter_util.quantity_radii_from_minimum_and_maximum_radii_and_radii_points(
50+
minimum_radius=0.0, maximum_radius=1.0, radii_points=4
51+
)
52+
53+
assert quantity_radii == [0.0, 0.25, 0.5, 0.75, 1.0]
54+
55+
quantity_radii = plotter_util.quantity_radii_from_minimum_and_maximum_radii_and_radii_points(
56+
minimum_radius=0.0, maximum_radius=2.0, radii_points=4
57+
)
58+
59+
assert quantity_radii == [0.0, 0.5, 1.0, 1.5, 2.0]
60+
61+
quantity_radii = plotter_util.quantity_radii_from_minimum_and_maximum_radii_and_radii_points(
62+
minimum_radius=1.0, maximum_radius=3.0, radii_points=4
63+
)
64+
65+
assert quantity_radii == [1.0, 1.5, 2.0, 2.5, 3.0]
66+
67+
def test__annuli_radii_from_minimum_and_maximum_radii__is_correct_values(self):
68+
quantity_radii, annuli_radii = plotter_util.quantity_and_annuli_radii_from_minimum_and_maximum_radii_and_radii_points(
69+
minimum_radius=0.0, maximum_radius=1.0, radii_points=2
70+
)
71+
72+
assert quantity_radii == [0.25, 0.75]
73+
assert annuli_radii == [0.0, 0.5, 1.0]
74+
75+
quantity_radii, annuli_radii = plotter_util.quantity_and_annuli_radii_from_minimum_and_maximum_radii_and_radii_points(
76+
minimum_radius=0.0, maximum_radius=1.0, radii_points=3
77+
)
78+
79+
assert quantity_radii == [(1.0 / 6.0), (3.0 / 6.0), (5.0 / 6.0)]
80+
assert annuli_radii == [0.0, (1.0 / 3.0), (2.0 / 3.0), 1.0]

0 commit comments

Comments
 (0)