def test_time_dimension_not_multiple_of_orig_time_dimension(self): # ten_year_dataset.times.size is 120 nt_avg = 11 # Temporal Rebin to exactly 10 (time) values dataset = dp.temporal_rebin_with_time_index( self.ten_year_dataset, nt_avg) new_times = self.ten_year_dataset.times[::11][:-1] self.assertEqual(dataset.times.size, self.ten_year_dataset.times.size / nt_avg) np.testing.assert_array_equal(dataset.times, new_times)
def test_time_dimension_not_multiple_of_orig_time_dimension(self): # ten_year_dataset.times.size is 120 nt_avg = 11 # Temporal Rebin to exactly 10 (time) values dataset = dp.temporal_rebin_with_time_index( self.ten_year_dataset, nt_avg) new_times = self.ten_year_dataset.times[::11][:-1] self.assertEqual(dataset.times.size, self.ten_year_dataset.times.size // nt_avg) np.testing.assert_array_equal(dataset.times, new_times)
def test_time_dimension_multiple_of_orig_time_dimension(self): # ten_year_dataset.times.size is 120 nt_avg = self.ten_year_dataset.times.size / 2 # Temporal Rebin to exactly 2 (time) values dataset = dp.temporal_rebin_with_time_index( self.ten_year_dataset, nt_avg) start_time = self.ten_year_dataset.times[0] # First month of the middle year middle_element = self.ten_year_dataset.times.size / 2 end_time = self.ten_year_dataset.times[middle_element] self.assertEqual(dataset.times.size, self.ten_year_dataset.times.size / nt_avg) np.testing.assert_array_equal(dataset.times, [start_time, end_time])
def test_time_dimension_multiple_of_orig_time_dimension(self): # ten_year_dataset.times.size is 120 nt_avg = self.ten_year_dataset.times.size // 2 # Temporal Rebin to exactly 2 (time) values dataset = dp.temporal_rebin_with_time_index( self.ten_year_dataset, nt_avg) start_time = self.ten_year_dataset.times[0] # First month of the middle year middle_element = self.ten_year_dataset.times.size // 2 end_time = self.ten_year_dataset.times[middle_element] self.assertEqual(dataset.times.size, self.ten_year_dataset.times.size // nt_avg) np.testing.assert_array_equal(dataset.times, [start_time, end_time])
def test_returned_dataset_attributes(self): nt_avg = 3 dataset = dp.temporal_rebin_with_time_index( self.ten_year_dataset, nt_avg) new_times = self.ten_year_dataset.times[::3] new_values = self.ten_year_dataset.values[::3] self.assertEqual(self.ten_year_dataset.name, dataset.name) self.assertEqual(self.ten_year_dataset.origin, dataset.origin) self.assertEqual(self.ten_year_dataset.units, dataset.units) self.assertEqual(self.ten_year_dataset.variable, dataset.variable) np.testing.assert_array_equal(new_times, dataset.times) np.testing.assert_array_equal(new_values, dataset.values) np.testing.assert_array_equal(self.ten_year_dataset.lats, dataset.lats) np.testing.assert_array_equal(self.ten_year_dataset.lons, dataset.lons)