Coverage for rivapy/tools/__init__.py: 100%

1# -*- coding: utf-8 -*- 

2from rivapy.tools.factory import _factory 

3from rivapy.tools.datetools import Period, Schedule 

4from rivapy.tools.datetime_grid import DateTimeGrid 

5from rivapy.tools.scheduler import SimpleSchedule, PeakSchedule, OffPeakSchedule, GasSchedule 

6 

7# __all__ = ['_converter', '_validators', 'datetools', 'enums'] 

8 

9 

10def _add_to_factory(cls): 

11 factory_entries = _factory() 

12 factory_entries[cls.__name__] = cls 

13 

14 

15_add_to_factory(SimpleSchedule) 

16_add_to_factory(PeakSchedule) 

17_add_to_factory(OffPeakSchedule) 

18_add_to_factory(GasSchedule)