Coverage for gpaw/tddft/solvers/__init__.py: 87%

1from .base import BaseSolver 

2from .bicgstab import BiCGStab 

3from .cscg import CSCG 

4 

5__all__ = ['create_solver', 'BiCGStab', 'CSCG'] 

6 

7 

8def create_solver(name, **kwargs): 

9 if isinstance(name, BaseSolver): 

10 return name 

11 elif isinstance(name, dict): 

12 kwargs.update(name) 

13 return create_solver(**kwargs) 

14 elif name == 'CSCG': 

15 return CSCG(**kwargs) 

16 elif name == 'BiCGStab': 

17 return BiCGStab(**kwargs) 

18 else: 

19 raise ValueError('Unknown solver: %s' % name)