|
1 | | -# (C) British Crown Copyright 2013 - 2017, Met Office |
| 1 | +# (C) British Crown Copyright 2013 - 2018, Met Office |
2 | 2 | # |
3 | 3 | # This file is part of Iris. |
4 | 4 | # |
@@ -101,18 +101,22 @@ def test_time_360(self): |
101 | 101 | time_coord = DimCoord([0, 100.1, 200.2, 300.3, 400.4], |
102 | 102 | long_name="time", units=time_unit) |
103 | 103 | cube.add_dim_coord(time_coord, 0) |
104 | | - if netCDF4.__version__ > '1.2.4': |
| 104 | + # TODO Remove this if statement as we move to cftime |
| 105 | + if cf_units.__version__ > '1': |
| 106 | + import cftime |
| 107 | + expected_index = [cftime.Datetime360Day(2000, 1, 1, 0, 0), |
| 108 | + cftime.Datetime360Day(2000, 4, 11, 2, 24), |
| 109 | + cftime.Datetime360Day(2000, 7, 21, 4, 48), |
| 110 | + cftime.Datetime360Day(2000, 11, 1, 7, 12), |
| 111 | + cftime.Datetime360Day(2001, 2, 11, 9, 36)] |
| 112 | + |
| 113 | + else: |
105 | 114 | expected_index = [netcdftime.Datetime360Day(2000, 1, 1, 0, 0), |
106 | 115 | netcdftime.Datetime360Day(2000, 4, 11, 2, 24), |
107 | 116 | netcdftime.Datetime360Day(2000, 7, 21, 4, 48), |
108 | 117 | netcdftime.Datetime360Day(2000, 11, 1, 7, 12), |
109 | 118 | netcdftime.Datetime360Day(2001, 2, 11, 9, 36)] |
110 | | - else: |
111 | | - expected_index = [netcdftime.datetime(2000, 1, 1, 0, 0), |
112 | | - netcdftime.datetime(2000, 4, 11, 2, 24), |
113 | | - netcdftime.datetime(2000, 7, 21, 4, 48), |
114 | | - netcdftime.datetime(2000, 11, 1, 7, 12), |
115 | | - netcdftime.datetime(2001, 2, 11, 9, 36)] |
| 119 | + |
116 | 120 | series = iris.pandas.as_series(cube) |
117 | 121 | self.assertArrayEqual(series, cube.data) |
118 | 122 | self.assertArrayEqual(series.index, expected_index) |
@@ -244,12 +248,16 @@ def test_time_360(self): |
244 | 248 | time_coord = DimCoord([100.1, 200.2], long_name="time", |
245 | 249 | units=time_unit) |
246 | 250 | cube.add_dim_coord(time_coord, 0) |
247 | | - if netCDF4.__version__ > '1.2.4': |
| 251 | + # TODO: Remove this if statement once we move to exclusive cftime. |
| 252 | + if cf_units.__version__ > '1': |
| 253 | + # cf_units depends upon cftime, so we can safely assume we |
| 254 | + # have it. |
| 255 | + import cftime |
| 256 | + expected_index = [cftime.Datetime360Day(2000, 4, 11, 2, 24), |
| 257 | + cftime.Datetime360Day(2000, 7, 21, 4, 48)] |
| 258 | + else: |
248 | 259 | expected_index = [netcdftime.Datetime360Day(2000, 4, 11, 2, 24), |
249 | 260 | netcdftime.Datetime360Day(2000, 7, 21, 4, 48)] |
250 | | - else: |
251 | | - expected_index = [netcdftime.datetime(2000, 4, 11, 2, 24), |
252 | | - netcdftime.datetime(2000, 7, 21, 4, 48)] |
253 | 261 | expected_columns = [0, 1, 2, 3, 4] |
254 | 262 | data_frame = iris.pandas.as_data_frame(cube) |
255 | 263 | self.assertArrayEqual(data_frame, cube.data) |
|
0 commit comments