def test_works_fire1(self): self.particle_tracker = track.ParticleTracker(**fire1_data_p) self.particle_tracker.save_particle_tracks() f = h5py.File('tests/data/tracking_output/ptracks_test_fire1.hdf5', 'r') expected_snum = np.arange(600, 490, -50) actual_snum = f['snum'][...] npt.assert_allclose(expected_snum, actual_snum) expected_id = np.array([24565150, 24079833, 13109563, 14147322]) actual_id = f['ID'][...] npt.assert_allclose(expected_id, actual_id) expected_rho_500 = np.array([ 2.98729116e-09, 3.12611002e-08, 8.95081308e-04, np.nan ]) * constants.UNITDENSITY_IN_NUMDEN actual_rho_500 = f['Den'][...][:, -1] npt.assert_allclose(expected_rho_500, actual_rho_500) expected_ptype_p0 = np.array([4, 0, 0]) actual_ptype_p0 = f['PType'][...][0] npt.assert_allclose(expected_ptype_p0, actual_ptype_p0)
def test_saves(self): # Check for existing output files and delete them. self.id_finder.drop_duplicates = True output_filepath = 'tests/data/tracking_output/ptracks_test.hdf5' if os.path.isfile(output_filepath): os.remove(output_filepath) # Update kwargs used_kwargs = copy.deepcopy(default_data_p) used_kwargs['custom_fns'] = test_apply_fns particle_tracker = track.ParticleTracker(**used_kwargs) # Run the function particle_tracker.save_particle_tracks() # Check results f = h5py.File(output_filepath, 'r') # The output for this function should be an integer assert np.issubdtype(f['IDx2'][...].dtype, np.integer) # None of the elements should be 0. assert not (0 in f['IDx2'][...]) # None of the elements should be 0. assert not (0. in f['Potential'][...])
def setUp(self): # Mock the code version so we don't repeatedly change test data patcher = patch('galaxy_dive.utils.utilities.get_code_version') self.addCleanup(patcher.stop) self.mock_code_version = patcher.start() kwargs = dict(default_data_p) kwargs['n_processors'] = 2 self.particle_tracker = track.ParticleTracker(**kwargs) # The name of the function. self.fn = self.particle_tracker.save_particle_tracks
def test_works_with_stars_early_on( self ): '''Test this works even in early snapshots, when there are no stars.''' self.particle_tracker = track.ParticleTracker( **early_star_data_p ) self.particle_tracker.save_particle_tracks() f = h5py.File( 'tests/data/tracking_output/ptracks_test_star_early.hdf5', 'r' ) expected_snum = np.array([ 11, 10 ]) actual_snum = f['snum'][...] npt.assert_allclose( expected_snum, actual_snum ) expected_id = np.array([2040268, 7909745, 8961984]) actual_id = f['ID'][...] npt.assert_allclose( expected_id, actual_id )
def setUp(self): # Mock the code version so we don't repeatedly change test data patcher = patch('galaxy_dive.utils.utilities.get_code_version') self.addCleanup(patcher.stop) self.mock_code_version = patcher.start() self.particle_tracker = track.ParticleTracker(**default_data_p) # The name of the function. self.fn = self.particle_tracker.save_particle_tracks # Check for existing output files and delete them. output_filepath = 'tests/data/tracking_output/ptracks_test.hdf5' if os.path.isfile(output_filepath): os.remove(output_filepath)
import linefinder.track as track kwargs = { 'sdir': './linefinder/tests/data/test_data_with_new_id_scheme', 'p_types': [ 0, ], 'snum_start': 500, 'snum_end': 600, 'snum_step': 50, 'out_dir': './linefinder/tests/data/tracking_output', 'ids_tag': 'test', 'tag': 'test_jug', } particle_tracker = track.ParticleTracker(**kwargs) particle_tracker.save_particle_tracks_jug()