def create_cluster(self):
self.cluster = Cluster(settings=self.settings)
self.cluster.create()
from __future__ import unicode_literals
import pytest
import os
from datasciencebox.core import salt
from datasciencebox.core.project import Project
from datasciencebox.core.cloud.cluster import Cluster
from datasciencebox.core.cloud.instance import Instance
cluster = Cluster()
cluster.instances.append(
Instance(ip='0.0.0.0', username='******', keypair='/home/ubuntu/.ssh/id_rsa'))
cluster.instances.append(
Instance(ip='1.1.1.1:2222',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa2'))
cluster.instances.append(
Instance(ip='2.2.2.2',
port='3333',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa3'))
project = Project()
project.cluster = cluster
head_roles = ['head', 'head2', 'conda']
compute_roles = ['minion2', 'conda']
salt.HEAD_ROLES = head_roles
salt.COMPUTE_ROLES = compute_roles
