def test_zap_map_kwargs(default_structure, default_simulator):
z_dict_no_beam = generate_zap_map(default_structure,
default_simulator,
with_direct_beam=False)
z_dict_yes_beam = generate_zap_map(default_structure,
default_simulator,
with_direct_beam=True)
for k in z_dict_no_beam.keys():
# both dictionary's have the same keys
assert k in z_dict_yes_beam.keys()
# no beam has one fewer spots than yes beam
assert (z_dict_no_beam[k].intensities.shape[0] ==
z_dict_yes_beam[k].intensities.shape[0] - 1)
def test_zap_map_density_changes(default_structure, default_simulator, density):
""" Checks density arguments are passed correctly """
z_dict = generate_zap_map(default_structure, default_simulator, density=density)
if density == "3":
assert str(len(z_dict.keys())) == "3"
elif density == "7":
assert len(z_dict.keys()) > 5 # monoclinic case gives 6 rather than 7
def test_zap_map_all_systems(default_structure, default_simulator, system):
z_dict = generate_zap_map(default_structure,
default_simulator,
system=system)
assert (0, 0, 1) in z_dict.keys()
assert (0, 0, 0) not in z_dict.keys()
