diff options
author | Mark Wiebe <mwiebe@enthought.com> | 2011-06-02 16:49:25 -0500 |
---|---|---|
committer | Mark Wiebe <mwiebe@enthought.com> | 2011-06-02 16:52:13 -0500 |
commit | 6c2c90c4a3bda6acb89e27d299d312f795e452bf (patch) | |
tree | 0a73c9cb2df7a69e04c3f0b82af54f261c7ce7c2 /numpy/core/tests | |
parent | e08fbd04ff4f6aa71d67d722eef4fa290894fd94 (diff) | |
download | numpy-6c2c90c4a3bda6acb89e27d299d312f795e452bf.tar.gz |
ENH: datetime: Eliminate use of npy_timedeltastruct, an unnecessary abstraction
Diffstat (limited to 'numpy/core/tests')
-rw-r--r-- | numpy/core/tests/test_datetime.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/numpy/core/tests/test_datetime.py b/numpy/core/tests/test_datetime.py index fe130fb05..1059aeaa1 100644 --- a/numpy/core/tests/test_datetime.py +++ b/numpy/core/tests/test_datetime.py @@ -99,6 +99,17 @@ class TestDateTime(TestCase): assert_equal(np.datetime64(b, 's'), a) assert_equal(np.datetime64(b, 's').dtype, np.dtype('M8[s]')) + # Construction from datetime.date + assert_equal(np.datetime64('1945-03-25'), + np.datetime64(datetime.date(1945,3,25))) + assert_equal(np.datetime64('2045-03-25', 'D'), + np.datetime64(datetime.date(2045,3,25), 'D')) + # Construction from datetime.datetime + assert_equal(np.datetime64('1980-01-25T14:36:22.5Z'), + np.datetime64(datetime.datetime(1980,1,25, + 14,36,22,500000))) + + def test_timedelta_scalar_construction(self): # Construct with different units assert_equal(np.timedelta64(7, 'D'), @@ -126,6 +137,26 @@ class TestDateTime(TestCase): assert_equal(np.timedelta64(b, 's'), a) assert_equal(np.timedelta64(b, 's').dtype, np.dtype('m8[s]')) + # Construction from datetime.timedelta + assert_equal(np.timedelta64(5, 'D'), + np.timedelta64(datetime.timedelta(days=5))) + assert_equal(np.timedelta64(102347621, 's'), + np.timedelta64(datetime.timedelta(seconds=102347621))) + assert_equal(np.timedelta64(-10234760000, 'us'), + np.timedelta64(datetime.timedelta( + microseconds=-10234760000))) + assert_equal(np.timedelta64(10234760000, 'us'), + np.timedelta64(datetime.timedelta( + microseconds=10234760000))) + assert_equal(np.timedelta64(1023476, 'ms'), + np.timedelta64(datetime.timedelta(milliseconds=1023476))) + assert_equal(np.timedelta64(10, 'm'), + np.timedelta64(datetime.timedelta(minutes=10))) + assert_equal(np.timedelta64(281, 'h'), + np.timedelta64(datetime.timedelta(hours=281))) + assert_equal(np.timedelta64(28, 'W'), + np.timedelta64(datetime.timedelta(weeks=28))) + def test_datetime_nat_casting(self): a = np.array('NaT', dtype='M8[D]') b = np.datetime64('NaT', '[D]') |