def test_from_filepath(request, tmpdir):
import os
testname = request.node.name
tempdir = tmpdir.mkdir("rootdir")
fpath = os.path.join(tempdir.strpath, "{}.yaml".format(testname))
cluster = Cluster("foo")
username = "******"
keypair = "~/.ssh/key"
n = 5
for i in range(n):
instance = Instance(uid="%i" % i,
ip="{0}.{0}.{0}.{0}".format(i),
username=username,
keypair=keypair)
cluster.append(instance)
cluster.to_file(fpath)
cluster2 = Cluster.from_filepath(fpath)
assert len(cluster2.instances) == n
assert len(cluster.instances) == len(cluster2.instances)
for i, instance in enumerate(cluster2.instances):
assert instance.uid == "%i" % i
assert instance.ip == "{0}.{0}.{0}.{0}".format(i)
assert instance.username == username
assert instance.keypair == keypair
def test_from_filepath(request, tmpdir):
import os
testname = request.node.name
tempdir = tmpdir.mkdir("rootdir")
fpath = os.path.join(tempdir.strpath, "{}.yaml".format(testname))
cluster = Cluster()
username = "******"
keypair="~/.ssh/key"
n = 5
for i in range(n):
instance = Instance(uid="%i" % i, ip="{0}.{0}.{0}.{0}".format(i), username=username, keypair=keypair)
cluster.append(instance)
cluster.to_file(fpath)
cluster2 = Cluster.from_filepath(fpath)
assert len(cluster2.instances) == n
assert len(cluster.instances) == len(cluster2.instances)
for i, instance in enumerate(cluster2.instances):
assert instance.uid == "%i" % i
assert instance.ip == "{0}.{0}.{0}.{0}".format(i)
assert instance.username == username
assert instance.keypair == keypair
def test_append_instance():
cluster = Cluster("foo")
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert len(cluster.instances) == n
assert cluster.head == cluster.instances[0]
def test_append_instance():
cluster = Cluster()
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert len(cluster.instances) == n
assert cluster.head == cluster.instances[0]
def test_set_keypair():
cluster = Cluster("foo")
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert cluster.instances[i].keypair is None
pkey = "ubuntu"
cluster.set_keypair(pkey)
for i in range(n):
assert cluster.instances[i].keypair == pkey
def test_set_username():
cluster = Cluster("foo")
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert cluster.instances[i].username is None
user = "******"
cluster.set_username(user)
for i in range(n):
assert cluster.instances[i].username == user
def test_set_keypair():
cluster = Cluster()
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert cluster.instances[i].keypair is None
pkey = "ubuntu"
cluster.set_keypair(pkey)
for i in range(n):
assert cluster.instances[i].keypair == pkey
def test_set_username():
cluster = Cluster()
n = 5
for i in range(n):
instance = Instance(ip="%i" % i)
cluster.append(instance)
assert cluster.instances[i].username is None
user = "******"
cluster.set_username(user)
for i in range(n):
assert cluster.instances[i].username == user
def test_dict_serde():
cluster = Cluster()
username = "******"
keypair="~/.ssh/key"
n = 5
for i in range(n):
instance = Instance(uid="%i" % i, ip="{0}.{0}.{0}.{0}".format(i), username=username, keypair=keypair)
cluster.append(instance)
data = cluster.to_dict()
cluster2 = Cluster.from_dict(data)
assert len(cluster2.instances) == n
for i, instance in enumerate(cluster2.instances):
assert instance.uid == "%i" % i
assert instance.ip == "{0}.{0}.{0}.{0}".format(i)
assert instance.username == username
assert instance.keypair == keypair
def test_dict_serde():
cluster = Cluster("foo")
username = "******"
keypair = "~/.ssh/key"
n = 5
for i in range(n):
instance = Instance(uid="%i" % i,
ip="{0}.{0}.{0}.{0}".format(i),
username=username,
keypair=keypair)
cluster.append(instance)
data = cluster.to_dict()
cluster2 = Cluster.from_dict(data)
assert len(cluster2.instances) == n
for i, instance in enumerate(cluster2.instances):
assert instance.uid == "%i" % i
assert instance.ip == "{0}.{0}.{0}.{0}".format(i)
assert instance.username == username
assert instance.keypair == keypair
def test_append_non_instance_type():
cluster = Cluster("foo")
with pytest.raises(DaskEc2Exception):
cluster.append({"wrong": "type"})
def test_append_non_instance_type():
cluster = Cluster()
with pytest.raises(DaskEc2Exception) as excinfo:
cluster.append({"wrong": "type"})
