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)
