def test_get_subset_for_symmetry_first_360():
"""Test that first 360 degrees are selected from each sweep"""
exclude_images, expts, tables = _make_input_for_exclude_tests(
exclude_images=False)
refls = get_subset_for_symmetry(expts, tables, exclude_images)
assert refls[0]["i"].all_eq(0)
assert refls[1]["i"].all_eq(0)
def test_get_subset_for_symmetry_image_range():
"""Test that first 360 degrees are selected from each sweep with an exclude
images command."""
exclude_images, expts, tables = _make_input_for_exclude_tests(exclude_images=True)
refls = get_subset_for_symmetry(expts, tables, exclude_images)
assert refls[0]["i"].all_eq(0)
assert refls[1]["i"].all_eq(0)
def test_get_subset_for_symmetry_prior_image_range():
"""Test that first 360 degrees are selected from each sweep with an exclude
images command."""
exclude_images, expts, tables = _make_input_for_exclude_tests(exclude_images=True)
# Explicitly exclude a different image range
expts = exclude_image_ranges_from_scans(tables, expts, [["0:1:360"], ["1:1:360"]])
refls = get_subset_for_symmetry(expts, tables)
assert refls[0]["i"].all_eq(1)
assert refls[1]["i"].all_eq(1)
# Exclude_images should be cumulative, i.e. the range exclude above should
# be excluded in addition to the new range provided explicitly to the
# function
refls = get_subset_for_symmetry(expts, tables, exclude_images)
assert len(refls[0]) == 0
assert len(refls[1]) == 0