def setUp(self): super().setUp() self.project = ProjectFactory() self.run = RunFactory( user=self.user, project=self.project, content="test", raw_content="test", is_managed=True, ) self.url = "/{}/{}/{}/runs/{}/artifacts_lineage/names/".format( API_V1, "polyaxon", self.project.name, self.run.uuid.hex) self.objects = [] for i in range(self.num_objects): obj = self.factory_class( name=f"foo{i}", state=uuid.uuid4(), ) self.objects.append(obj) ArtifactLineage.objects.create(run=self.run, artifact=obj) self.query = self.queryset.filter(run=self.run) # Other user objects other_project = ProjectFactory() other_run = RunFactory( user=self.user, project=other_project, content="test", raw_content="test", is_managed=True, ) other_obj = self.factory_class(state=uuid.uuid4(), ) ArtifactLineage.objects.create(run=other_run, artifact=other_obj)
def setUp(self): super().setUp() self.project = ProjectFactory() self.url = "/{}/polyaxon/{}/runs/".format(API_V1, self.project.name) self.objects = [ self.factory_class(project=self.project, user=self.user) for _ in range(self.num_objects) ] self.queryset = self.model_class.objects.filter( project=self.project).order_by("-updated_at") # one object that does not belong to the filter self.other_project = ProjectFactory() self.other_url = "/{}/polyaxon/{}/runs/".format( API_V1, self.other_project.name) self.other_object = self.factory_class(project=self.other_project, )
def setUp(self): super().setUp() self.project = ProjectFactory() self.objects = [] self.objects.append( self.factory_class( project=self.project, user=self.user, tags=["tag11"], ) ) self.objects.append( self.factory_class( project=self.project, user=self.user, tags=["new", "tag21", "tag22"], ) ) self.objects.append(self.factory_class(project=self.project, user=self.user)) self.objects.append(self.factory_class(project=self.project, user=self.user)) self.url = "/{}/{}/{}/runs/tag/".format( API_V1, self.user.username, self.project.name ) self.queryset = self.model_class.objects.filter(project=self.project).order_by( "created_at" )
def setUp(self): super().setUp() self.project = ProjectFactory() self.run = RunFactory( project=self.project, content="test", raw_content="test", is_managed=True, ) self.state = self.project.owner.uuid
def setUp(self): super().setUp() # Force tasks autodiscover from scheduler import tasks # noqa self.user = UserFactory() self.project = ProjectFactory()
def setUp(self): super().setUp() self.project = ProjectFactory() self.objects = [ self.factory_class(project=self.project, user=self.user) for _ in range(4) ] self.url = "/{}/{}/{}/runs/stop/".format(API_V1, self.user.username, self.project.name)
def setUp(self): super().setUp() self.user = UserFactory() self.user2 = UserFactory() self.project = ProjectFactory() op_spec = OperationSpecification.read(values=get_fxt_job_with_inputs()) self.run = compile_operation_run(project_id=self.project.id, op_spec=op_spec, user_id=self.user.id)
def setUp(self): super().setUp() self.project = ProjectFactory() self.objects = [ self.factory_class( project=self.project, user=self.user, pending=V1RunPending.APPROVAL ) for _ in range(4) ] self.url = "/{}/{}/{}/runs/approve/".format( API_V1, self.user.username, self.project.name )
def setUp(self): super().setUp() self.project = ProjectFactory() self.object = self.factory_class( user=self.user, project=self.project, content="test", raw_content="test", is_managed=True, ) self.url = "/{}/{}/{}/runs/{}/".format(API_V1, self.user.username, self.project.name, self.object.uuid.hex) self.queryset = self.model_class.objects.all()
def setUp(self): super().setUp() self.project = ProjectFactory() self.objects = [ self.factory_class(project=self.project, user=self.user) for _ in range(4) ] for obj in self.objects: new_run_status( run=obj, condition=V1StatusCondition.get_condition( type=V1Statuses.RUNNING, status=True), ) self.url = "/{}/{}/{}/runs/stop/".format(API_V1, self.user.username, self.project.name)
def setUp(self): super().setUp() self.project = ProjectFactory() self.run = RunFactory( user=self.user, project=self.project, content="test", raw_content="test", is_managed=True, ) self.artifact = self.factory_class(name="foo", state=self.project.owner.uuid) self.artifact_lineage = ArtifactLineage.objects.create( artifact=self.artifact, run=self.run) self.url = "/{}/{}/{}/runs/{}/artifacts_lineage/{}/".format( API_V1, "polyaxon", self.project.name, self.run.uuid.hex, self.artifact.name, ) self.queryset = Artifact.objects.all()
def setUp(self): super().setUp() self.project = ProjectFactory() self.objects = [ RunFactory(project=self.project, user=self.user) for _ in range(3) ] self.url = "/{}/polyaxon/{}/runs/artifacts_lineage/".format( API_V1, self.project.name, ) obj = self.factory_class(name="in1", state=self.project.uuid) ArtifactLineage.objects.create(run=self.objects[0], artifact=obj) ArtifactLineage.objects.create(run=self.objects[1], artifact=obj) obj = self.factory_class(name="in2", state=self.project.uuid) ArtifactLineage.objects.create(run=self.objects[0], artifact=obj) obj = self.factory_class(name="out1", state=self.project.uuid) ArtifactLineage.objects.create( run=self.objects[0], artifact=obj, is_input=False, ) self.query = self.queryset.filter(run__project=self.project)
def setUp(self): super().setUp() self.user = UserFactory() self.project = ProjectFactory() self.run = RunFactory(project=self.project)
def setUp(self): super().setUp() self.project = ProjectFactory() self.url = "/{}/polyaxon/{}/runs/".format(API_V1, self.project.name)