def test_no(self): # Collapse all data into one number data, output = self._get() x, y, _, _, _ = output._get_x_y(data, verif.axis.No()) self.assertEqual(1, len(x)) self.assertEqual(1, y.shape[0]) self.assertEqual(2, y.shape[1])
def test_leadtime(self): data, output = self._get() x, y, _, _, _ = output._get_x_y(data, verif.axis.Leadtime()) lt = data.leadtimes np.testing.assert_array_equal(x, lt) self.assertEqual(2, y.shape[1]) self.assertEqual(len(lt), y.shape[0])
def test_location(self): data, output = self._get() for ax in [verif.axis.Location(), verif.axis.Lat(), verif.axis.Lon(), verif.axis.Elev()]: x, y, _, _ = output._get_x_y(data, verif.axis.Location()) loc = data.locations self.assertEqual(len(loc), len(x)) self.assertEqual(2, y.shape[1]) self.assertEqual(len(loc), y.shape[0])
def test_year(self): data, output = self._get() x, y, _, _, _ = output._get_x_y(data, verif.axis.Year()) # There should only be one year in this dataset self.assertEqual(1, len(x)) self.assertEqual(1, y.shape[0]) # The date represents the first day of the year: self.assertEqual(20120101, verif.util.datenum_to_date(x[0])) self.assertEqual(2, y.shape[1])
def test_time(self): data, output = self._get() x, y, _, _, _ = output._get_x_y(data, verif.axis.Time()) # 20120101 to 20120301 self.assertEqual(61, len(x)) self.assertEqual(61, y.shape[0]) self.assertEqual(20120101, verif.util.datenum_to_date(x[0])) self.assertEqual(20120102, verif.util.datenum_to_date(x[1])) self.assertEqual(20120301, verif.util.datenum_to_date(x[-1])) self.assertEqual(2, y.shape[1])
def test_month(self): data, output = self._get() x, y, _, _, _ = output._get_x_y(data, verif.axis.Month()) # There should be three months in this dataset self.assertEqual(3, len(x)) self.assertEqual(3, y.shape[0]) # The dates represents the first day of the months: self.assertEqual(20120101, verif.util.datenum_to_date(x[0])) self.assertEqual(20120201, verif.util.datenum_to_date(x[1])) self.assertEqual(20120301, verif.util.datenum_to_date(x[2])) self.assertEqual(2, y.shape[1])