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()