def generate_module():
RANDOM_STATE = np.random.RandomState(1031412)
fname = os.path.join(os.path.dirname(__file__), "..", "src", "metrics", "test.rs")
module = Module(imports=["metrics::ranking::*"])
for num in range(5):
name = "roc_auc_test_%s" % num
module.add_test(_generate_test(name, RANDOM_STATE))
return fname, module
def generate_module():
RANDOM_STATE = np.random.RandomState(1031412)
fname = os.path.join(os.path.dirname(__file__), '..', 'src', 'metrics',
'test.rs')
module = Module(imports=['metrics::ranking::*'])
for num in range(5):
name = 'roc_auc_test_%s' % num
module.add_test(_generate_test(name, RANDOM_STATE))
return fname, module
def onEnableChannel(self, channel):
Module.onEnableChannel(self, channel)
if channel == 'ups_downs':
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
elif channel == 'tracker_status':
self.addStatusKey('state')
self.addStatusKey('tracker_status')
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
elif channel == 'downloads':
self.addStatusKey('eta')
self.addStatusKey('total_done')
self.addStatusKey('total_size')
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
def onEnableChannel(self, channel):
Module.onEnableChannel(self, channel)
if channel == 'ups_downs':
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
elif channel == 'tracker_status':
self.addStatusKey('state')
self.addStatusKey('tracker_status')
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
elif channel == 'downloads':
self.addStatusKey('eta')
self.addStatusKey('total_done')
self.addStatusKey('total_size')
self.addStatusKey('download_payload_rate')
self.addStatusKey('upload_payload_rate')
def __init__(self, cfg=dict()):
Module.__init__(self, 'deluge')
self.dc = None
self.statusKeys = []
self.result = None
self.interval = cfg.get('interval', 15)
self.ups = 0
self.downs = 0
self.trackers = {}
self.downloads = {}
self.torrents = []
self.channels = {
'ups_downs': False,
'trackers': False,
'downloads': False
}
self.svc = None
def __init__(self, cfg=dict()):
Module.__init__(self, 'deluge')
self.dc = None
self.statusKeys = []
self.result = None
self.interval = cfg.get('interval', 15)
self.ups = 0
self.downs = 0
self.trackers = {}
self.downloads = {}
self.torrents = []
self.channels = {
'ups_downs': False,
'trackers': False,
'downloads': False
}
self.svc = None
