Coverage for python/gsfit/database_writers/tokamak_energy_mdsplus_new/__init__.py: 89%

1from typing import TYPE_CHECKING 

2 

3from ..interface import DatabaseWriterProtocol 

4from ..workflow_names import workflow_names_from_settings 

5from .map_results_to_database import map_results_to_database 

6 

7if TYPE_CHECKING: 

8 from ...gsfit import Gsfit 

9 

10 

11class DatabaseWriterTokamakEnergyMDSplusNew(DatabaseWriterProtocol): 

12 def map_results_to_database(self, gsfit_controller: "Gsfit") -> None: 

13 return map_results_to_database(self, gsfit_controller) 

14 

15 def get_workflow_names(self, gsfit_controller: "Gsfit") -> list[str]: 

16 return workflow_names_from_settings(gsfit_controller.settings)