@@ -39,18 +39,16 @@ def test_set_distances_at_angle(inputs, expected):
39
39
40
40
41
41
params3 = [
42
- ([1 , 3 , 1 , 45 ], [1 , 0.2431167 , 0.2431167 , 0.135335 , 0.135335 ]),
43
- ([1 , 3 , 1 , 90 ], [1 , 1 , 0.135335 , 0.135335 , 0.135335 ]),
44
- ([1 , 3 , 1 , 120 ], [1 , 1 , 0.135335 , 0.049787 , 0.176921 ]),
42
+ ([1 ], [1 , 1 , 0.135335 , 0.049787 , 0.176921 ]),
43
+ ([2 ], [1 , 1 , 0.018316 , 0.002479 , 0.031301 ]),
45
44
]
46
45
47
46
48
47
@pytest .mark .parametrize ("inputs, expected" , params3 )
49
48
def test_set_muls_at_angle (inputs , expected ):
50
49
expected_muls = expected
51
- actual_gs = Gridded_circle (radius = inputs [0 ], n_points_on_diameter = inputs [1 ], mu = inputs [2 ])
52
- actual_gs .set_distances_at_angle (inputs [3 ])
53
- actual_gs .set_muls_at_angle (inputs [3 ])
50
+ actual_gs = Gridded_circle (radius = 1 , n_points_on_diameter = 3 , mu = inputs [0 ])
51
+ actual_gs .set_muls_at_angle (120 )
54
52
actual_muls_sorted = sorted (actual_gs .muls )
55
53
expected_muls_sorted = sorted (expected_muls )
56
54
assert actual_muls_sorted == pytest .approx (expected_muls_sorted , rel = 1e-4 , abs = 1e-6 )
0 commit comments