def testSeaCIFunc(self):
'''Use Sea object to test mean and function passing'''
time = list(range(200))
window = dt.timedelta(minutes=3)
delta = dt.timedelta(minutes=1)
with warnings.catch_warnings(record=True) as w:
warnings.filterwarnings('always', 'Window size changed .*',
UserWarning, '^spacepy\\.seapy$')
compobj1 = seapy.Sea(self.unidata, time, self.epochs, \
window=window, delta=delta,verbose=False)
compobj1.sea(ci=95, ci_quan=np.mean)
compobj2 = seapy.Sea(self.unidata, time, self.epochs, \
window=window, delta=delta,verbose=False)
compobj2.sea(ci=95, ci_quan='mean')
ntest.assert_allclose(compobj1.bound_low, compobj2.bound_low)
def testRestoreEpochs(self):
"""Check that restoreepochs fails with no bad epochs"""
with warnings.catch_warnings(record=True) as w:
warnings.filterwarnings('always', 'Window size changed .*',
UserWarning, '^spacepy\\.seapy$')
self.obj = seapy.Sea(self.unidata, self.time, \
self.epochs, verbose=False)
re_fun = self.obj.restoreepochs
self.assertRaises(AttributeError, re_fun)
def setUp(self):
"""Setup block executed for each test in this class"""
super(SEATestsUniform, self).setUp()
self.testval = 5
self.unidata = [self.testval]*200
time = list(range(200))
self.epochs = [20,40,60,80,100,120,140,160,180]
with warnings.catch_warnings():
warnings.filterwarnings('always', 'Window size changed .*',
UserWarning, '^spacepy\\.seapy$')
self.obj = seapy.Sea(self.unidata, time, self.epochs, verbose=False)
self.obj.sea()
def setUp(self):
"""Setup block executed for each test in this class"""
super(SEATestsUniWithBad, self).setUp()
self.testval = 5
self.unidata = [self.testval]*200
#insert badvals
for ind in range(30,180,16):
self.unidata[ind] = -99
time = list(range(200))
self.epochs = [20,40,60,80,100,120,140,160,180]
self.obj = seapy.Sea(self.unidata, time, self.epochs, verbose=False)
self.obj.sea(badval=-99)
def testRandomBeforeSea(self):
n_req = 27
sttime = dt.datetime(2010, 1, 1)
time = [sttime + dt.timedelta(minutes=x) for x in range(200)]
epochs = [sttime + dt.timedelta(minutes=x) for x in self.epochs]
window = dt.timedelta(minutes=3)
delta = dt.timedelta(minutes=1)
with warnings.catch_warnings(record=True) as w:
warnings.filterwarnings('always', 'Window size changed .*',
UserWarning, '^spacepy\\.seapy$')
compobj = seapy.Sea(self.unidata, time, epochs, \
window=window, delta=delta,verbose=False)
newsea = compobj.random(n_req)
self.assertEqual(len(newsea), n_req)
def testDatetimeEquiv(self):
"""Test of equivalence of serial and datetime handling"""
sttime = dt.datetime(2010, 1, 1)
time = [sttime + dt.timedelta(minutes=x) for x in range(200)]
epochs = [sttime + dt.timedelta(minutes=x) for x in self.epochs]
window = dt.timedelta(minutes=3)
delta = dt.timedelta(minutes=1)
with warnings.catch_warnings(record=True) as w:
warnings.filterwarnings('always', 'Window size changed .*',
UserWarning, '^spacepy\\.seapy$')
compobj = seapy.Sea(self.unidata, time, epochs, \
window=window, delta=delta,verbose=False)
compobj.sea()
ntest.assert_array_equal(self.obj.semedian, compobj.semedian)
ntest.assert_array_equal(self.obj.semean, compobj.semean)
