def test_load_component_info_to_2d_geometry_irregular(geom_file):
geometry = mantid.load_component_info_to_2d(geom_file,
sizes={
'y': 2,
'x': 50
})
assert geometry["position"].sizes == {'x': 50, 'y': 2}
assert "x" in geometry
assert "y" in geometry
def test_load_component_info_to_2d_geometry_advanced_geom(geom_file):
geometry = mantid.load_component_info_to_2d(geom_file,
sizes={
'x': 10,
'y': 10
},
advanced_geometry=True)
assert geometry["position"].sizes == {'x': 10, 'y': 10}
assert geometry["rotation"].sizes == {'x': 10, 'y': 10}
assert geometry["shape"].sizes == {'x': 10, 'y': 10}
def test_load_component_info_to_2d_geometry_non_cartesian(geom_file):
geometry = mantid.load_component_info_to_2d(geom_file,
sizes={
'u': 20,
'v': 5
})
assert geometry["position"].sizes == {'u': 20, 'v': 5}
# Cannot extract x, y as fields
assert "x" not in geometry
assert "y" not in geometry
def test_load_component_info_to_2d_geometry(geom_file):
geometry = mantid.load_component_info_to_2d(geom_file,
sizes={
'x': 10,
'y': 10
})
assert geometry["position"].sizes == {'x': 10, 'y': 10}
assert sc.identical(
geometry["x"],
sc.DataArray(data=sc.array(
dims=["x"], values=np.arange(0.0, 0.1, 0.01), unit=sc.units.m)))
assert sc.identical(
geometry["y"],
sc.DataArray(data=sc.array(
dims=["y"], values=np.arange(0.0, 0.1, 0.01), unit=sc.units.m)))
def test_load_component_info_to_2d_geometry_bad_sizes(geom_file):
bad_sizes = {'x': 10, 'y': 5} # gives volume of 50 not 100
with pytest.raises(ValueError):
mantid.load_component_info_to_2d(geom_file, sizes=bad_sizes)