Python temporal_rebin_with_time_index Examples

Example #1

File: test_dataset_processor.py Project: ahujamoh/climate

 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)

Example #2

 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)

Example #3

File: test_dataset_processor.py Project: ahujamoh/climate

 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])

Example #4

 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])

Example #5

File: test_dataset_processor.py Project: ahujamoh/climate

 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)

Example #6

 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)