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"})