5252import logging
5353import pytest
5454
55- from powerapi .cli .tools import CommonCLIParser , generate_pushers , generate_pullers
55+ from powerapi .cli .tools import CommonCLIParser , PusherGenerator , PullerGenerator
5656from powerapi .backendsupervisor import BackendSupervisor
5757from powerapi .formula import DummyFormulaActor
5858from powerapi .dispatch_rule import HWPCDispatchRule , HWPCDepthLevel
@@ -133,14 +133,15 @@ def test_run(files, supervisor):
133133
134134 config = {'verbose' : LOG_LEVEL ,
135135 'stream' : False ,
136- 'input' : [ {'files' : FILES ,
137- 'model' : 'hwpc_report' ,
138- 'name' : 'puller' ,
139- 'type' : 'csv' }] ,
140- 'output' : [{ ' model' : 'power_report' , 'name' : 'pusher' ,'type' : 'csv' , ' directory' : ROOT_PATH }] }
136+ 'input' : { 'csv' : {'files' : FILES ,
137+ 'model' : 'hwpc_report' ,
138+ 'name' : 'puller' ,
139+ }} ,
140+ 'output' : { 'csv' : { ' model' : 'power_report' , 'name' : 'pusher' , ' directory' : ROOT_PATH }} }
141141
142142 # Pusher
143- pushers = generate_pushers (config )
143+ pusher_generator = PusherGenerator ()
144+ pushers = pusher_generator .generate (config )
144145
145146 # Formula
146147 formula_factory = (lambda name ,
@@ -157,7 +158,8 @@ def test_run(files, supervisor):
157158 # Puller
158159 report_filter = Filter ()
159160 report_filter .filter (lambda msg : True , dispatcher )
160- pullers = generate_pullers (config , report_filter )
161+ puller_generator = PullerGenerator (report_filter )
162+ pullers = puller_generator .generate (config )
161163
162164 for _ , pusher in pushers .items ():
163165 supervisor .launch_actor (pusher )
0 commit comments