Coverage for gpaw/lcaotddft/observer.py: 88%

1from gpaw.analyse.observers import Observer 

2 

3 

4class TDDFTObserver(Observer): 

5 

6 def __init__(self, paw, interval): 

7 super().__init__(interval) 

8 self.timer = paw.timer 

9 if hasattr(paw, 'time') and hasattr(paw, 'niter'): 

10 paw.attach(self, interval, paw) 

11 

12 def update(self, paw): 

13 self.timer.start('%s update' % self.__class__.__name__) 

14 self._update(paw) 

15 self.timer.stop('%s update' % self.__class__.__name__) 

16 

17 def _update(self, paw): 

18 raise NotImplementedError() 

19 

20 def write_restart(self): 

21 """Write restart file. 

22 

23 Optional method that will be called by RestartFileWriter 

24 when writing the calculator restart file. 

25 """ 

26 

27 def __str__(self): 

28 return self.__class__.__name__