def public_source_group2(public_group2):
obj = ObjFactory(groups=[public_group2])
source = Source(obj_id=obj.id, group_id=public_group2.id)
DBSession.add(source)
DBSession.commit()
yield obj
ObjFactory.teardown(obj)
def public_source_two_groups(public_group, public_group2):
obj = ObjFactory(groups=[public_group, public_group2])
sources = []
for group in [public_group, public_group2]:
source = Source(obj_id=obj.id, group_id=group.id)
sources.append(source)
DBSession.add(source)
DBSession.commit()
yield obj
ObjFactory.teardown(obj)
def public_candidate(public_filter, user):
obj = ObjFactory(groups=[public_filter.group])
candidate = Candidate(
obj=obj,
filter=public_filter,
passed_at=datetime.utcnow() - timedelta(seconds=np.random.randint(0, 100)),
uploader_id=user.id,
)
DBSession.add(candidate)
DBSession.commit()
yield obj
ObjFactory.teardown(obj)
def public_candidate_two_groups(public_filter, public_filter2, public_group,
public_group2, user):
obj = ObjFactory(groups=[public_group, public_group2])
candidates = []
for filter_ in [public_filter, public_filter2]:
candidate = Candidate(
obj=obj,
filter=filter_,
passed_at=datetime.utcnow() -
timedelta(seconds=np.random.randint(0, 100)),
uploader_id=user.id,
)
candidates.append(candidate)
DBSession.add(candidate)
DBSession.commit()
yield obj
ObjFactory.teardown(obj)
def private_source():
obj = ObjFactory(groups=[])
yield obj
ObjFactory.teardown(obj)
def public_obj(public_group):
obj = ObjFactory(groups=[public_group])
yield obj
# If the obj wasn't deleted by the test using it, clean up
ObjFactory.teardown(obj)
def public_ZTFe028h94k(public_group):
obj = ObjFactory(groups=[public_group], ra=229.9620403, dec=34.8442757)
DBSession().add(Source(obj_id=obj.id, group_id=public_group.id))
DBSession().commit()
yield obj
ObjFactory.teardown(obj)
def public_ZTF21aaeyldq(public_group):
obj = ObjFactory(groups=[public_group], ra=123.813909, dec=-5.867007)
DBSession().add(Source(obj_id=obj.id, group_id=public_group.id))
DBSession().commit()
yield obj
ObjFactory.teardown(obj)
def public_ZTF20acgrjqm(public_group):
obj = ObjFactory(groups=[public_group], ra=65.0630767, dec=82.5880983)
DBSession().add(Source(obj_id=obj.id, group_id=public_group.id))
DBSession().commit()
yield obj
ObjFactory.teardown(obj)