From 338bb14f7f7b65dcccb41c8828094053be5fb408 Mon Sep 17 00:00:00 2001 From: Zebedee Nicholls Date: Tue, 2 Mar 2021 08:48:47 +1100 Subject: [PATCH] Add failing test --- cf_units/tests/unit/unit/test_Unit.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/cf_units/tests/unit/unit/test_Unit.py b/cf_units/tests/unit/unit/test_Unit.py index 201e845c..fd942a4c 100644 --- a/cf_units/tests/unit/unit/test_Unit.py +++ b/cf_units/tests/unit/unit/test_Unit.py @@ -75,11 +75,12 @@ def test_gregorian_calendar_conversion_shape(self): self.assertEqual(expected.shape, result.shape) def test_non_gregorian_calendar_conversion_dtype(self): - data = np.arange(4, dtype=np.float32) - u1 = Unit('hours since 2000-01-01 00:00:00', calendar='360_day') - u2 = Unit('hours since 2000-01-02 00:00:00', calendar='360_day') - result = u1.convert(data, u2) - self.assertEqual(result.dtype, np.float32) + for start_dtype in [np.float32, np.float64]: + data = np.arange(4, dtype=start_dtype) + u1 = Unit('hours since 2000-01-01 00:00:00', calendar='360_day') + u2 = Unit('hours since 2000-01-02 00:00:00', calendar='360_day') + result = u1.convert(data, u2) + self.assertEqual(result.dtype, start_dtype) class Test_convert__endianness_time(unittest.TestCase):