def setUpClass(cls): cls.view_id = None cls.group_id, cls.project_id, cls.session_id, cls.acquisition_id = create_test_acquisition() # Get user id user = cls.fw.get_current_user() cls.user_id = user.id cls.session = cls.fw.get_session(cls.session_id) cls.subject = cls.session.subject
def setUp(self): self.group_id, self.project_id, self.session_id, self.acquisition_id = create_test_acquisition( ) self.gear_id = create_test_gear()
def setUp(self): self.group_id, self.project_id, self.session_id, self.acquisition_id = create_test_acquisition() self.tmpfile = None
def setUp(self): self.group_id, self.project_id, self.subject_id, self.session_id, self.acquisition_id = create_test_acquisition( return_subject=True) self.gear_id = None
def createDestination(self): self.group_id, self.project_id, self.session_id, self.acquisition_id = create_test_acquisition( ) self.context.destination['id'] = self.acquisition_id
def test_collections(self): fw = self.fw collection_name = self.rand_string() collection = flywheel.Collection(label=collection_name, description=self.rand_string()) # Add self.collection_id = collection_id = fw.add_collection(collection) self.assertNotEmpty(collection_id) # Get saved_collection = fw.get_collection(collection_id) self.assertEqual(saved_collection.id, collection_id) self.assertEqual(saved_collection.label, collection_name) self.assertEqual(saved_collection.description, collection.description) self.assertTimestampBeforeNow(saved_collection.created) self.assertGreaterEqual(saved_collection.modified, saved_collection.created) # Generic Get is equivalent self.assertEqual( fw.get(collection_id).to_dict(), saved_collection.to_dict()) # Add acquisition to the collection group_id, project_id, session_id, acquisition_id = create_test_acquisition( ) self.group_ids.append(group_id) self.project_ids.append(project_id) saved_collection.add_acquisitions(acquisition_id) # Get sessions saved_sessions = saved_collection.sessions() self.assertEqual(len(saved_sessions), 1) self.assertEqual(saved_sessions[0].id, session_id) # Get acquisitions saved_acquisitions = saved_collection.acquisitions() self.assertEqual(len(saved_acquisitions), 1) self.assertEqual(saved_acquisitions[0].id, acquisition_id) # Get session acquisitions saved_session_acquisitions = fw.get_collection_acquisitions( collection_id, session=session_id) self.assertEqual(len(saved_session_acquisitions), 1) self.assertEqual(saved_session_acquisitions[0].id, acquisition_id) # Add session to the collection group_id, project_id, session_id2, acquisition_id2 = create_test_acquisition( ) self.group_ids.append(group_id) self.project_ids.append(project_id) saved_collection.add_sessions(session_id2) saved_collection = saved_collection.reload() # Get sessions saved_sessions = saved_collection.sessions() self.assertEqual(len(saved_sessions), 2) self.assertEqual( len(list(filter(lambda x: x.id == session_id, saved_sessions))), 1) self.assertEqual( len(list(filter(lambda x: x.id == session_id2, saved_sessions))), 1) # Get acquisitions saved_acquisitions = saved_collection.acquisitions() self.assertEqual(len(saved_acquisitions), 2) self.assertEqual( len( list( filter(lambda x: x.id == acquisition_id, saved_acquisitions))), 1) self.assertEqual( len( list( filter(lambda x: x.id == acquisition_id2, saved_acquisitions))), 1) # Get session acquisitions saved_session_acquisitions = fw.get_collection_acquisitions( collection_id, session=session_id) self.assertEqual(len(saved_session_acquisitions), 1) self.assertEqual(saved_session_acquisitions[0].id, acquisition_id) # Get All saved_collection = saved_collection.reload() # Clear internal state collections = fw.get_all_collections() self.sanitize_for_collection(saved_collection, info_exists=False) self.assertIn(saved_collection, collections) # Modify new_name = self.rand_string() saved_collection.update(label=new_name) changed_collection = saved_collection.reload() self.assertEqual(changed_collection.label, new_name) self.assertEqual(changed_collection.created, saved_collection.created) self.assertGreater(changed_collection.modified, saved_collection.modified) # Add note message = 'This is a note' saved_collection.add_note(message) tag = 'example-tag' saved_collection.add_tag(tag) # Replace Info saved_collection.replace_info({'foo': 3, 'bar': 'qaz'}) # Set Info saved_collection.update_info({'foo': 42, 'hello': 'world'}) # Check changed_collection = saved_collection.reload() self.assertEqual(len(changed_collection.notes), 1) self.assertEqual(changed_collection.notes[0].text, message) self.assertEqual(len(changed_collection.tags), 1) self.assertEqual(changed_collection.tags[0], tag) self.assertEqual(changed_collection.info['foo'], 42) self.assertEqual(changed_collection.info['bar'], 'qaz') self.assertEqual(changed_collection.info['hello'], 'world') # Delete info fields saved_collection.delete_info(['foo', 'bar']) changed_collection = saved_collection.reload() self.assertNotIn('foo', changed_collection.info) self.assertNotIn('bar', changed_collection.info) self.assertEqual(changed_collection.info['hello'], 'world') # Delete fw.delete_collection(collection_id) self.collection_id = None collections = fw.get_all_collections() self.sanitize_for_collection(changed_collection) self.assertNotIn(changed_collection, collections)