Coverage for /builds/kinetik161/ase/ase/visualize/__init__.py: 66.67%

1import ase.parallel as parallel 

2 

3 

4def view(atoms, data=None, viewer='ase', repeat=None, block=False, **kwargs): 

5 from ase.visualize.viewers import VIEWERS 

6 

7 if parallel.world.size > 1: 

8 return 

9 

10 vwr = VIEWERS[viewer.lower()] 

11 handle = vwr.view(atoms, data=data, repeat=repeat, **kwargs) 

12 

13 if block and hasattr(handle, 'wait'): 

14 status = handle.wait() 

15 if status != 0: 

16 raise RuntimeError(f'Viewer "{vwr.name}" failed with status ' 

17 '{status}') 

18 

19 return handle