def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a list of BCT adapters; imports a CFF data-set """ self.test_user = TestFactory.create_user("BCT_User") self.test_project = TestFactory.create_project(self.test_user, "BCT-Project") ### Make sure Connectivity is in DB TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = dao.get_generic_entity(Connectivity, 'John Doe', 'subject')[0] # make weights matrix symmetric, or else some BCT algorithms will run infinitely: w = self.connectivity.weights self.connectivity.weights = w + w.T - numpy.diag(w.diagonal()) self.algo_groups = dao.get_generic_entity(model.AlgorithmGroup, 'MatlabAdapter', 'classname') self.assertTrue(self.algo_groups is not None) self.assertEquals(6, len(self.algo_groups)) self.bct_adapters = [] for group in self.algo_groups: self.bct_adapters.append( TestFactory.create_adapter(group, self.test_project))
def test_project_disk_size(self): project1 = TestFactory.create_project(self.test_user, 'test_proj1') zip_path = os.path.join(os.path.dirname(tvb_data.__file__), 'connectivity', 'connectivity_66.zip') TestFactory.import_zip_connectivity(self.test_user, project1, 'testSubject', zip_path) project2 = TestFactory.create_project(self.test_user, 'test_proj2') TestFactory.import_cff(test_user=self.test_user, test_project=project2) projects = self.project_service.retrieve_projects_for_user( self.test_user.id)[0] self.assertNotEqual(projects[0].disk_size, projects[1].disk_size, "projects should have different size") for project in projects: self.assertNotEqual(0, project.disk_size) self.assertNotEqual('0.0 KiB', project.disk_size_human) prj_folder = self.structure_helper.get_project_folder(project) actual_disk_size = self.compute_recursive_h5_disk_usage( prj_folder)[0] ratio = float(actual_disk_size) / project.disk_size msg = "Real disk usage: %s The one recorded in the db : %s" % ( actual_disk_size, project.disk_size) self.assertTrue(ratio < 1.4, msg)
def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a list of BCT adapters; imports a CFF data-set """ self.test_user = TestFactory.create_user("BCT_User") self.test_project = TestFactory.create_project(self.test_user, "BCT-Project") ### Make sure Connectivity is in DB TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = dao.get_generic_entity(Connectivity, 'John Doe', 'subject')[0] # make weights matrix symmetric, or else some BCT algorithms will run infinitely: w = self.connectivity.weights self.connectivity.weights = w + w.T - numpy.diag(w.diagonal()) algorithms = dao.get_generic_entity(model.Algorithm, 'Brain Connectivity Toolbox', 'group_description') self.assertTrue(algorithms is not None) self.assertTrue(len(algorithms) > 5) self.bct_adapters = [] for algo in algorithms: self.bct_adapters.append(ABCAdapter.build_adapter(algo))
def setUp(self): """ Sets up the environment for running the tests; creates a test user and a test project, saves old configuration and imports a CFF data-set """ self.test_user = TestFactory.create_user() self.test_project = TestFactory.create_project(self.test_user) TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project)
def test_bad_reference(self): TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) field = FilterChain.datatype + '.subject' filters = FilterChain('', [field], [TEST_SUBJECT_A], ['!=']) bad_reference_connectivity = TestFactory.get_entity(self.test_project, Connectivity(), filters) self.assertRaises(OperationException, self._import_csv_test_connectivity, bad_reference_connectivity.gid, TEST_SUBJECT_A)
def setUp(self): """ Sets up the environment for running the tests; creates a test user and a test project, saves old configuration and imports a CFF data-set """ self.test_user = TestFactory.create_user() self.test_project = TestFactory.create_project(self.test_user) TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project)
def test_bad_reference(self): TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) field = FilterChain.datatype + '.subject' filters = FilterChain('', [field], [TEST_SUBJECT_A], ['!=']) bad_reference_connectivity = TestFactory.get_entity( self.test_project, Connectivity(), filters) self.assertRaises(OperationException, self._import_csv_test_connectivity, bad_reference_connectivity.gid, TEST_SUBJECT_A)
def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a surface; imports a CFF data-set """ self.datatypeFactory = DatatypesFactory() self.test_project = self.datatypeFactory.get_project() self.test_user = self.datatypeFactory.get_user() TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = TestFactory.get_entity(self.test_project, Connectivity()) self.assertTrue(self.connectivity is not None)
def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a surface; imports a CFF data-set """ self.datatypeFactory = DatatypesFactory() self.test_project = self.datatypeFactory.get_project() self.test_user = self.datatypeFactory.get_user() TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = TestFactory.get_entity(self.test_project, Connectivity()) self.assertTrue(self.connectivity is not None)
def setUp(self): """ Reset the database before each test. """ self.import_service = ImportService() self.flow_service = FlowService() self.project_service = ProjectService() self.test_user = TestFactory.create_user() self.test_project = TestFactory.create_project(self.test_user, name="GeneratedProject", description="test_desc") self.operation = TestFactory.create_operation(test_user=self.test_user, test_project=self.test_project) self.adapter_instance = TestFactory.create_adapter(test_project=self.test_project) TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.zip_path = None
def setUp(self): """ Reset the database before each test. """ self.import_service = ImportService() self.flow_service = FlowService() self.project_service = ProjectService() self.test_user = TestFactory.create_user() self.test_project = TestFactory.create_project(self.test_user, name="GeneratedProject", description="test_desc") self.operation = TestFactory.create_operation( test_user=self.test_user, test_project=self.test_project) self.adapter_instance = TestFactory.create_adapter() TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.zip_path = None
def test_project_disk_size(self): project1 = TestFactory.create_project(self.test_user, 'test_proj1') zip_path = os.path.join(os.path.dirname(tvb_data.__file__), 'connectivity', 'connectivity_66.zip') TestFactory.import_zip_connectivity(self.test_user, project1, 'testSubject', zip_path) project2 = TestFactory.create_project(self.test_user, 'test_proj2') TestFactory.import_cff(test_user=self.test_user, test_project=project2) projects = self.project_service.retrieve_projects_for_user(self.test_user.id)[0] self.assertNotEqual(projects[0].disk_size, projects[1].disk_size, "projects should have different size") for project in projects: self.assertNotEqual(0, project.disk_size) self.assertNotEqual('0.0 KiB', project.disk_size_human) prj_folder = self.structure_helper.get_project_folder(project) actual_disk_size = self.compute_recursive_h5_disk_usage(prj_folder)[0] ratio = float(actual_disk_size) / project.disk_size msg = "Real disk usage: %s The one recorded in the db : %s" % (actual_disk_size, project.disk_size) self.assertTrue(ratio < 1.4, msg)
def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a list of BCT adapters; imports a CFF data-set """ self.test_user = TestFactory.create_user("BCT_User") self.test_project = TestFactory.create_project(self.test_user, "BCT-Project") ### Make sure Connectivity is in DB TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = dao.get_generic_entity(Connectivity, 'John Doe', 'subject')[0] # make weights matrix symmetric, or else some BCT algorithms will run infinitely: w = self.connectivity.weights self.connectivity.weights = w + w.T - numpy.diag(w.diagonal()) algorithms = dao.get_generic_entity(model.Algorithm, 'Brain Connectivity Toolbox', 'group_description') self.assertTrue(algorithms is not None) self.assertTrue(len(algorithms) > 5) self.bct_adapters = [] for algo in algorithms: self.bct_adapters.append(ABCAdapter.build_adapter(algo))
def setUp(self): """ Sets up the environment for running the tests; creates a test user, a test project, a connectivity and a list of BCT adapters; imports a CFF data-set """ self.test_user = TestFactory.create_user("BCT_User") self.test_project = TestFactory.create_project(self.test_user, "BCT-Project") ### Make sure Connectivity is in DB TestFactory.import_cff(test_user=self.test_user, test_project=self.test_project) self.connectivity = dao.get_generic_entity(Connectivity, 'John Doe', 'subject')[0] # make weights matrix symmetric, or else some BCT algorithms will run infinitely: w = self.connectivity.weights self.connectivity.weights = w + w.T - numpy.diag(w.diagonal()) self.algo_groups = dao.get_generic_entity(model.AlgorithmGroup, 'MatlabAdapter', 'classname') self.assertTrue(self.algo_groups is not None) self.assertEquals(6, len(self.algo_groups)) self.bct_adapters = [] for group in self.algo_groups: self.bct_adapters.append(TestFactory.create_adapter(group, self.test_project))