1919def cartesian_geometry ():
2020 coordinates = np .linspace (- 1 ,1 ,5 )
2121 all_coords = np .array (np .meshgrid (coordinates ,coordinates ,coordinates ,indexing = 'ij' )).reshape (3 ,- 1 ).T
22- # Scale so they are different lengths
23- all_coords *= np .array ((1.1 ,1.0 ,0.9 ))
2422 node_ids = np .arange (all_coords .shape [0 ])+ 1
2523 rotation = sdpy .rotation .R (2 ,20 ,degrees = True )@sdpy .rotation .R (1 ,- 30 ,degrees = True )@sdpy .rotation .R (0 ,- 45 ,degrees = True )
2624 translation = np .array (((0.0 ,2.0 ,1.0 ),))
@@ -98,6 +96,12 @@ def test_node_indexing(cartesian_geometry):
9896 scalar_node = cartesian_geometry .node (scalar_id )
9997 check_nodes = cartesian_geometry .node [indices ]
10098 check_scalar_node = cartesian_geometry .node [scalar_index ]
99+ nodes_by_reduction = cartesian_geometry .node .reduce (ids )
100+ scalar_node_by_reduction = cartesian_geometry .node .reduce (scalar_id )
101101 assert np .all (nodes == check_nodes )
102102 assert np .all (check_scalar_node == scalar_node )
103+ assert np .all (nodes_by_reduction == check_nodes )
104+ assert np .all (check_scalar_node == scalar_node_by_reduction )
105+ with pytest .raises (ValueError ):
106+ cartesian_geometry .node (1000 )
103107
0 commit comments