def test_store_load_complete_region_mapping(tmph5factory, connectivity_factory, surface_factory, region_mapping_factory): connectivity = connectivity_factory(2) surface = surface_factory(5) region_mapping = region_mapping_factory(surface, connectivity) with ConnectivityH5(tmph5factory('Connectivity_{}.h5'.format(connectivity.gid))) as conn_h5: conn_h5.store(connectivity) conn_stored = Connectivity() conn_h5.load_into(conn_stored) with SurfaceH5(tmph5factory('Surface_{}.h5'.format(surface.gid))) as surf_h5: surf_h5.store(surface) surf_stored = Surface() surf_h5.load_into(surf_stored) with RegionMappingH5(tmph5factory('RegionMapping_{}.h5'.format(region_mapping.gid))) as rm_h5: rm_h5.store(region_mapping) rm_stored = RegionMapping() rm_h5.load_into(rm_stored) # load_into will not load dependent datatypes. connectivity and surface are undefined with pytest.raises(TraitAttributeError): rm_stored.connectivity with pytest.raises(TraitAttributeError): rm_stored.surface rm_stored.connectivity = conn_stored rm_stored.surface = surf_stored assert rm_stored.connectivity is not None assert rm_stored.surface is not None
def test_store_load_connectivity(tmph5factory, connectivity_factory): connectivity = connectivity_factory(2) conn_h5 = ConnectivityH5(tmph5factory()) conn_h5.store(connectivity) conn_h5.close() conn_stored = Connectivity() with pytest.raises(TraitAttributeError): conn_stored.region_labels conn_h5.load_into(conn_stored) assert conn_stored.region_labels.shape[0] == 2
def test_store_partial_connectivity(tmph5factory): partial_conn = Connectivity( region_labels=numpy.array(["a", "b"]), weights=numpy.zeros((2, 2)), tract_lengths=numpy.zeros((2, 2)), centres=numpy.zeros((2, 2)), number_of_regions=int(2), number_of_connections=int(4), ) conn_h5 = ConnectivityH5(tmph5factory()) conn_h5.store(partial_conn) conn_h5.close()
def test_store_connectivity(tmph5factory, connectivity_factory): connectivity = connectivity_factory(2) conn_h5 = ConnectivityH5(tmph5factory()) conn_h5.store(connectivity) conn_h5.close()