Merge pull request #18 from yucongalicechen/muls

Tests for setting muls

Author: sbillinge

src/diffpy/labpdfproc/tests/test_functions.py

@@ -36,3 +36,19 @@ def test_set_distances_at_angle(inputs, expected):
     actual_distances_sorted = sorted(actual_gs.distances)
     expected_distances_sorted = sorted(expected_distances)
     assert actual_distances_sorted == pytest.approx(expected_distances_sorted, rel=1e-4, abs=1e-6)
+
+
+params3 = [
+    ([1], [1, 1, 0.135335, 0.049787, 0.176921]),
+    ([2], [1, 1, 0.018316, 0.002479, 0.031301]),
+]
+
+
+@pytest.mark.parametrize("inputs, expected", params3)
+def test_set_muls_at_angle(inputs, expected):
+    expected_muls = expected
+    actual_gs = Gridded_circle(radius=1, n_points_on_diameter=3, mu=inputs[0])
+    actual_gs.set_muls_at_angle(120)
+    actual_muls_sorted = sorted(actual_gs.muls)
+    expected_muls_sorted = sorted(expected_muls)
+    assert actual_muls_sorted == pytest.approx(expected_muls_sorted, rel=1e-4, abs=1e-6)