def test_registerRunner(): with pytest.raises(RunnerClassNameError): PyPPL.registerRunner(object) class RunnerX: pass PyPPL.registerRunner(RunnerX) assert PyPPL.RUNNERS['x'] is RunnerX assert PyPPL.RUNNERS['local'] is runner.RunnerLocal assert PyPPL.RUNNERS['dry'] is runner.RunnerDry assert PyPPL.RUNNERS['ssh'] is runner.RunnerSsh assert PyPPL.RUNNERS['sge'] is runner.RunnerSge assert PyPPL.RUNNERS['slurm'] is runner.RunnerSlurm
def testRegisterRunner(self, runner, name): if name in PyPPL.RUNNERS: del PyPPL.RUNNERS[name] PyPPL.registerRunner(runner) self.assertIs(PyPPL.RUNNERS[name], runner)