Coverage for rivapy / marketdata_tools / __init__.py: 92%

1from rivapy.marketdata_tools.factory import _factory 

2from rivapy.marketdata_tools.pfc_shifter import PFCShifter 

3from rivapy.marketdata_tools.pfc_shaper import PFCShaper, CategoricalRegression, SimpleCategoricalRegression, CategoricalFourierShaper 

4 

5 

6def _add_to_factory(cls): 

7 factory_entries = _factory() 

8 factory_entries[cls.__name__] = cls 

9 

10 

11_add_to_factory(PFCShifter) 

12_add_to_factory(PFCShaper) 

13_add_to_factory(CategoricalRegression) 

14_add_to_factory(SimpleCategoricalRegression) 

15_add_to_factory(CategoricalFourierShaper) 

16 

17 

18if __name__ == "__main__": 

19 pass