def test_init_both_cluster_and_id(self): _id = "test1" original_cluster = CustomSLURMCluster() with self.assertRaises(AssertionError): @on_cluster(cluster_id=_id, cluster=original_cluster) def f(): pass controller.delete_cluster(original_cluster.name) original_cluster = CustomSLURMCluster() @on_cluster(cluster_id=original_cluster.name, cluster=original_cluster) def f(): pass cluster, client = controller.get_cluster(id_=original_cluster.name) self.assertEqual(original_cluster, cluster) self.assertEqual(cluster.name, original_cluster.name) self.assertIsInstance(client, Client) controller.delete_cluster(original_cluster.name) original_cluster = CustomSLURMCluster(name=_id) @on_cluster(cluster_id=_id, cluster=original_cluster) def f(): pass cluster, client = controller.get_cluster(id_=_id) self.assertEqual(original_cluster, cluster) self.assertEqual(cluster.name, _id) self.assertIsInstance(client, Client) controller.delete_cluster(_id)
def test_init_blank(self): @on_cluster() def f(): pass cluster, client = controller.get_cluster() self.assertIsInstance(cluster, self.cluster_type) self.assertIsInstance(client, Client)
def _test_default_behavior(self, cluster_type): set_default_cluster(cluster_type) @on_cluster() def f(): pass cluster, client = controller.get_cluster() self.assertIsInstance(cluster, cluster_type) self.assertIsInstance(client, Client)
def test_init_only_id_local(self): _id = "test1" original_cluster = LocalCluster() controller.add_cluster(id_=_id, cluster=original_cluster) @on_cluster(cluster_id=_id) def f(): pass cluster, client = controller.get_cluster(id_=_id) self.assertEqual(original_cluster, cluster) self.assertIsInstance(client, Client) controller.delete_cluster(_id)
def test_init_only_cluster(self): original_cluster = CustomSLURMCluster() @on_cluster(cluster=original_cluster) def f(): pass cluster, client = controller.get_cluster(original_cluster.name) self.assertEqual(original_cluster, cluster) self.assertIsInstance(client, Client) controller.delete_cluster(cluster.name) _id = "test1" original_cluster = CustomSLURMCluster(name=_id) @on_cluster(cluster=original_cluster) def f(): pass cluster, client = controller.get_cluster(_id) self.assertEqual(original_cluster, cluster) self.assertIsInstance(client, Client) controller.delete_cluster(cluster.name)
def test_init_only_id(self): _id = "test1" original_cluster = CustomSLURMCluster(name=_id) with self.assertRaises(ClusterException) as ctx: controller.add_cluster(cluster=original_cluster) self.assertEqual('Cluster "{}" already exists!'.format(_id), str(ctx.exception)) @on_cluster(cluster_id=_id) def f(): pass cluster, client = controller.get_cluster(id_=_id) self.assertEqual(original_cluster, cluster) self.assertEqual(cluster.name, _id) self.assertIsInstance(client, Client) controller.delete_cluster(_id)
def test_init_only_cluster_local(self): original_cluster = LocalCluster() with self.assertRaises(ClusterException) as ctx: @on_cluster(cluster=original_cluster) def f(): pass self.assertEqual('LocalCluster requires "cluster_id" argument.', str(ctx.exception)) _id = "test1" @on_cluster(cluster=original_cluster, cluster_id=_id) def f(): pass cluster, client = controller.get_cluster(_id) self.assertEqual(original_cluster, cluster) self.assertIsInstance(client, Client) controller.delete_cluster(_id)
def _basic_tests(self, cluster_type, id_=None): cluster, client = controller.get_cluster(id_) self.assertIsInstance(cluster, cluster_type) self.assertIsInstance(client, Client) self.assertEqual(controller.default_cluster.__class__, cluster_type.__class__)
def test_init(self): cluster, client = clusters_controller_singleton.get_cluster() self.assertIsInstance(cluster, self.cluster_type) self.assertIsInstance(client, Client)