def test_get_run_info(self): trial = Trial() trial.add_resource(Resource('ra', data='some_data')) trial.add_resource(Resource('rb')) task_1 = Task([trial], plugin_1, 'plugin_category') self.assertTrue(task_1.is_ready) result = task_1.get_run_info() self.assertFalse(task_1.is_ready) self.assertEqual(len(task_1.locking_keys.keys()), 2) self.assertTrue(task_1.provides[0] in task_1.locking_keys.keys()) self.assertTrue(task_1.provides[1] in task_1.locking_keys.keys())
def test_get_args(self): '''_get_args checks out 'requirements' and returns them in a list.''' trial = Trial() trial.add_resource(Resource('ra', data='some_data')) trial.add_resource(Resource('rb')) task_1 = Task([trial], plugin_1, 'plugin_category') self.assertTrue(task_1.is_ready) result = task_1._get_args() self.assertTrue(task_1.is_ready) self.assertTrue('some_data' in result) self.assertTrue(trial.rb.data in result) self.assertEqual(len(result), 2)
def __init__(self, requires, provides): self.name = 'Faux Plugin' self.requires = requires self.provides = provides self.is_stochastic = False self.is_pooling = False self.unpool_as = None self.silent_pooling = False plugin_1 = FauxPlugin(requires=['ra', 'rb'], provides=['pb', 'pa']) plugin_2 = FauxPlugin(requires=['rb'], provides=['ra', 'pa']) trial_1 = Trial() trial_1.add_resource(Resource('ra', data='some_data')) trial_1.add_resource(Resource('rb')) class TaskTests(unittest.TestCase): def setUp(self): pass def test_constructor_1(self): '''Resources created on trial by task if plugin produces them.''' trial = Trial() self.assertFalse(hasattr(trial, 'pa')) self.assertFalse(hasattr(trial, 'pb')) task_1 = Task([trial], plugin_1, 'plugin_category') self.assertTrue(hasattr(trial, 'pa')) self.assertTrue(isinstance(trial.pa, Resource))