def test_Query_datetime1(self):
d = cf.Data([[1., 5.], [6, 2]],
'days since 2000-12-29 21:00:00',
calendar='standard')
message = 'Diff =' + str(
(d -
cf.Data(cf.dt('2001-01-03 21:00:00', calendar='standard'))).array)
self.assertTrue((d == cf.eq(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, True], [False, False]]), verbose=2), message)
self.assertTrue((d == cf.ne(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, False], [True, True]])), message)
self.assertTrue((d == cf.ge(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, True], [True, False]])), message)
self.assertTrue((d == cf.gt(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, False], [True, False]])), message)
self.assertTrue((d == cf.le(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, True], [False, True]])), message)
self.assertTrue((d == cf.lt(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, False], [False, True]])), message)
self.assertTrue((d == cf.wi(cf.dt('2000-12-31 21:00:00'),
cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, True],
[False, True]])), message)
self.assertTrue((d == cf.wo(cf.dt('2000-12-31 21:00:00'),
cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, False],
[True, False]])), message)
self.assertTrue((d == cf.set([
cf.dt('2000-12-31 21:00:00'),
cf.dt('2001-01-03 21:00:00')
])).equals(cf.Data([[False, True], [False, True]])), message)
_ = cf.seasons()
[
cf.seasons(n, start) for n in [1, 2, 3, 4, 6, 12]
for start in range(1, 13)
]
with self.assertRaises(Exception):
cf.seasons(13)
with self.assertRaises(Exception):
cf.seasons(start=8.456)
_ = cf.mam()
_ = cf.djf()
_ = cf.jja()
_ = cf.son()
def test_Query_year_month_day_hour_minute_second(self):
d = cf.Data([[1., 5.], [6, 2]], 'days since 2000-12-29 21:57:57',
calendar='gregorian')
self.assertTrue(
(d == cf.year(2000)).equals(
cf.Data([[True, False], [False, True]]))
)
self.assertTrue(
(d == cf.month(12)).equals(
cf.Data([[True, False], [False, True]]))
)
self.assertTrue(
(d == cf.day(3)).equals(cf.Data([[False, True], [False, False]])))
d = cf.Data([[1., 5], [6, 2]], 'hours since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.hour(2)).equals(cf.Data([[False, True], [False, False]])))
d = cf.Data([[1., 5], [6, 2]], 'minutes since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.minute(2)).equals(
cf.Data([[False, True], [False, False]]))
)
d = cf.Data([[1., 5], [6, 2]], 'seconds since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.second(2)).equals(
cf.Data([[False, True], [False, False]]))
)
d = cf.Data([[1., 5.], [6, 2]], 'days since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.year(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
self.assertTrue(
(d == cf.month(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
self.assertTrue(
(d == cf.day(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
d = cf.Data([[1., 5], [6, 2]], 'hours since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.hour(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
d = cf.Data([[1., 5], [6, 2]], 'minutes since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.minute(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
d = cf.Data([[1., 5], [6, 2]], 'seconds since 2000-12-29 21:57:57')
self.assertTrue(
(d == cf.second(cf.ne(-1))).equals(
cf.Data([[True, True], [True, True]]))
)
def test_Query_dteq_dtne_dtge_dtgt_dtle_dtlt(self):
d = cf.Data([[1., 5.], [6, 2]], 'days since 2000-12-29 21:00:00')
message = 'Diff =' + str((d-cf.Data(
cf.dt('2001-01-03 21:00:00', calendar='standard'))).array)
self.assertTrue(
(d == cf.eq(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, True], [False, False]])),
message
)
self.assertTrue(
(d == cf.ne(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, False], [True, True]])),
message
)
self.assertTrue(
(d == cf.ge(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, True], [True, False]])),
message
)
self.assertTrue(
(d == cf.gt(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[False, False], [True, False]])),
message
)
self.assertTrue(
(d == cf.le(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, True], [False, True]])),
message
)
self.assertTrue(
(d == cf.lt(cf.dt('2001-01-03 21:00:00'))).equals(
cf.Data([[True, False], [False, True]])),
message
)
self.assertTrue(
(d == cf.eq(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, True], [False, False]])),
message
)
self.assertTrue(
(d == cf.ne(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, False], [True, True]])),
message
)
self.assertTrue(
(d == cf.ge(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, True], [True, False]])),
message
)
self.assertTrue(
(d == cf.gt(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, False], [True, False]])),
message
)
self.assertTrue(
(d == cf.le(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, True], [False, True]])),
message
)
self.assertTrue(
(d == cf.lt(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, False], [False, True]])),
message
)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.eq(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.eq(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.eq(cf.dt('2100-1-1')), d)
self.assertNotEqual(
cf.eq(cf.dt('2001-1-1')) & cf.eq(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.ge(cf.dt(1990, 1, 1)), d)
self.assertEqual(cf.ge(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.ge(cf.dt('2100-1-1')), d)
self.assertNotEqual(
cf.ge(cf.dt('2001-1-1')) & cf.ge(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.gt(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.gt(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.gt(cf.dt('2100-1-1')), d)
self.assertEqual(
cf.gt(cf.dt('2001-1-1')) & cf.le(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.ne(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.ne(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.ne(cf.dt('2100-1-1')), d)
self.assertEqual(
cf.ne(cf.dt('2001-1-1')) & cf.ne(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertNotEqual(cf.le(cf.dt(1990, 1, 1)), d)
self.assertEqual(cf.le(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.le(cf.dt('2100-1-1')), d)
self.assertNotEqual(
cf.le(cf.dt('2001-1-1')) & cf.le(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertNotEqual(cf.lt(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.lt(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.lt(cf.dt('2100-1-1')), d)
self.assertNotEqual(
cf.lt(cf.dt('2001-1-1')) & cf.lt(cf.dt(2010, 12, 31)), d)
def test_Query_dteq_dtne_dtge_dtgt_dtle_dtlt(self):
d = cf.Data([[1.0, 5.0], [6, 2]], "days since 2000-12-29 21:00:00")
message = "Diff =" + str(
(d -
cf.Data(cf.dt("2001-01-03 21:00:00", calendar="standard"))).array)
self.assertTrue(
(d == cf.eq(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[False, True], [False, False]])),
message,
)
self.assertTrue(
(d == cf.ne(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[True, False], [True, True]])),
message,
)
self.assertTrue(
(d == cf.ge(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[False, True], [True, False]])),
message,
)
self.assertTrue(
(d == cf.gt(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[False, False], [True, False]])),
message,
)
self.assertTrue(
(d == cf.le(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[True, True], [False, True]])),
message,
)
self.assertTrue(
(d == cf.lt(cf.dt("2001-01-03 21:00:00"))).equals(
cf.Data([[True, False], [False, True]])),
message,
)
self.assertTrue(
(d == cf.eq(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, True], [False, False]])),
message,
)
self.assertTrue(
(d == cf.ne(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, False], [True, True]])),
message,
)
self.assertTrue(
(d == cf.ge(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, True], [True, False]])),
message,
)
self.assertTrue(
(d == cf.gt(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[False, False], [True, False]])),
message,
)
self.assertTrue(
(d == cf.le(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, True], [False, True]])),
message,
)
self.assertTrue(
(d == cf.lt(cf.dt(2001, 1, 3, 21, 0, 0))).equals(
cf.Data([[True, False], [False, True]])),
message,
)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.eq(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.eq(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.eq(cf.dt("2100-1-1")), d)
self.assertNotEqual(
cf.eq(cf.dt("2001-1-1")) & cf.eq(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.ge(cf.dt(1990, 1, 1)), d)
self.assertEqual(cf.ge(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.ge(cf.dt("2100-1-1")), d)
self.assertNotEqual(
cf.ge(cf.dt("2001-1-1")) & cf.ge(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.gt(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.gt(cf.dt(2002, 6, 16)), d)
self.assertNotEqual(cf.gt(cf.dt("2100-1-1")), d)
self.assertEqual(
cf.gt(cf.dt("2001-1-1")) & cf.le(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertEqual(cf.ne(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.ne(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.ne(cf.dt("2100-1-1")), d)
self.assertEqual(
cf.ne(cf.dt("2001-1-1")) & cf.ne(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertNotEqual(cf.le(cf.dt(1990, 1, 1)), d)
self.assertEqual(cf.le(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.le(cf.dt("2100-1-1")), d)
self.assertNotEqual(
cf.le(cf.dt("2001-1-1")) & cf.le(cf.dt(2010, 12, 31)), d)
d = cf.dt(2002, 6, 16)
self.assertNotEqual(cf.lt(cf.dt(1990, 1, 1)), d)
self.assertNotEqual(cf.lt(cf.dt(2002, 6, 16)), d)
self.assertEqual(cf.lt(cf.dt("2100-1-1")), d)
self.assertNotEqual(
cf.lt(cf.dt("2001-1-1")) & cf.lt(cf.dt(2010, 12, 31)), d)