Coverage for gpaw/new/poisson.py: 95%

1from math import nan 

2 

3 

4class PoissonSolver: 

5 def solve(self, 

6 vHt, 

7 rhot) -> float: 

8 raise NotImplementedError 

9 

10 def dipole_layer_correction(self) -> float: 

11 raise NotImplementedError 

12 

13 

14class PoissonSolverWrapper(PoissonSolver): 

15 def __init__(self, solver): 

16 self.description = solver.get_description() 

17 self.solver = solver 

18 

19 def __str__(self): 

20 return self.description 

21 

22 def solve(self, 

23 vHt, 

24 rhot) -> float: 

25 self.solver.solve(vHt.data, rhot.data) 

26 return nan 

27 

28 def dipole_layer_correction(self) -> float: 

29 try: 

30 return self.solver.correction 

31 except AttributeError: 

32 raise NotImplementedError