# -*- coding:utf-8 -*- """ Created on 5. aug. 2010 @author: pab """ import unittest from numpy.testing import TestCase, assert_array_almost_equal import wafo.data import wafo.objects as wo import wafo.spectrum.models as sm import wafo.transform.models as tm class TestTimeSeries(TestCase): def setUp(self): x = wafo.data.sea() self.ts = wo.mat2timeseries(x) def test_sampling_period(self): ts = self.ts assert_array_almost_equal(ts.sampling_period(), 0.25) def test_tospecdata(self): S = self.ts.tospecdata(L=150) print(S.data[:10].tolist()) assert_array_almost_equal(S.data[:10], [0.0050789888306202345, 0.0049411187454784225, 0.004553923924951667, 0.003990722577978725, 0.00335482379127744, 0.002755110296973988, 0.002281782794825119, 0.0019941282234629933, 0.0019329154962902488, 0.002164040256079313]) # [0.00913087, 0.00881073, 0.00791944, # 0.00664244, 0.00522429, 0.00389816, # 0.00282753, 0.00207843, 0.00162678, # 0.0013916]) def test_tocovdata(self): rf = self.ts.tocovdata(lag=150) assert_array_almost_equal(rf.data[:10], [0.22368637, 0.20838473, 0.17110733, 0.12237803, 0.07024054, 0.02064859, -0.02218831, -0.0555993, -0.07859847, -0.09166187]) def test_timeseries_trdata(self): Hs = 7.0 Sj = sm.Jonswap(Hm0=Hs) S = Sj.tospecdata() # Make spectrum object from numerical values S.tr = tm.TrOchi(mean=0, skew=0.16, kurt=0, sigma=Hs/4, ysigma=Hs/4) xs = S.sim(ns=2**20, iseed=10) ts = wo.mat2timeseries(xs) g0, _gemp = ts.trdata(monitor=False) # Not Monitor the development # Equal weight on all points g1, _gemp = ts.trdata(method='mnonlinear', gvar=0.5) # Less weight on the ends g2, _gemp = ts.trdata(method='nonlinear', gvar=[3.5, 0.5, 3.5]) self.assert_(1.2 < S.tr.dist2gauss() < 1.6) self.assert_(1.65 < g0.dist2gauss() < 2.05) self.assert_(0.54 < g1.dist2gauss() < 0.95) self.assert_(1.5 < g2.dist2gauss() < 1.9) def test_timeseries_wave_periods(self): true_t = ([-0.69, -0.86, -1.05], [0.42, 0.78, 1.37], [0.09, 0.51, -0.85], [-0.27, -0.08, 0.32], [3.84377468, 6.35707656, 4.15490909], [6.25273295, 6.10295202, 3.36978685], [2.48364668, 4.74282402, 1.75553431], [3.76908628, 1.360128, 1.61425254], [-5.05027968, -9.16405436, -15.60113092], [7.53392635, 13.90687837, 17.35666522], [-0.2811934, -7.11392635, -13.12687837], [4.05027968, 8.47405436, 14.74113092], [2.03999996, 0.07, 0.05], [-0.93, -0.07, -0.12], [1.10999996, 0., -0.07], [-0.86, -0.02, 0.3], [0.93, -0.8, -0.2], [1.10999996, 0., -0.07], [-0.02, 0.3, -0.34], [6.10295202, 3.36978685, 3.58501107], [6.25273295, 6.10295202, 3.36978685], ) pdefs = ['t2c', 'c2t', 't2t', 'c2c', 'd2d', 'u2u', 'd2u', 'u2d', 'd2t', 't2u', 'u2c', 'c2d', 'm2M', 'M2m', 'm2m', 'M2M', 'all', ] ts = wo.TimeSeries(self.ts.data[0:400, :2], self.ts.args[:400]) for pdef, truth in zip(pdefs, true_t): T, _ix = ts.wave_periods(vh=0.0, pdef=pdef) # print(T[:3,]) assert_array_almost_equal(T[:3], truth) true_t2 = ([1.10999996, 0., - 0.07], [-0.02, 0.3, - 0.34], [6.10295202, 3.369787, 3.585011], [6.25273295, 6.102952, 3.369787], [-0.27, -0.08, 0.32], [-0.27, -0.08, 0.32]) wdefs = ['mw', 'Mw', 'dw', 'uw', 'tw', 'cw', ] for wdef, truth in zip(wdefs, true_t2): pdef = '{0}2{0}'.format(wdef[0].lower()) T, _ix = ts.wave_periods(vh=0.0, pdef=pdef, wdef=wdef) print(T[:3]) assert_array_almost_equal(T[:3], truth) class TestObjects(TestCase): def setUp(self): x = wafo.data.sea() self.ts = wo.mat2timeseries(x) def test_cycles_and_levelcrossings(self): tp = self.ts.turning_points() assert_array_almost_equal(tp.data[:10], [-1.200495, 0.839505, -0.090495, -0.020495, -0.090495, -0.040495, -0.160495, 0.259505, -0.430495, -0.080495] ) mm = tp.cycle_pairs() assert_array_almost_equal(mm.data[:10], [0.839505, -0.020495, -0.040495, 0.259505, -0.080495, -0.080495, 0.349505, 0.859505, 0.009505, 0.319505]) true_lcs = (([0., 1., 2., 2., 3., 4., 5., 6., 7., 9.], [-1.7504945, -1.4404945, -1.4204945, -1.4004945, -1.3704945, -1.3204945, -1.2704945, -1.2604945, -1.2504945, -1.2004945]), ([0., 1., 2., 3., 3., 4., 5., 6., 7., 9.], [-1.7504945, -1.4404945, -1.4204945, -1.4004945, -1.3704945, -1.3204945, -1.2704945, -1.2604945, -1.2504945, -1.2004945]), ([1., 2., 3., 4., 4., 5., 6., 7., 9., 11.], [-1.7504945, -1.4404945, -1.4204945, -1.4004945, -1.3704945, -1.3204945, -1.2704945, -1.2604945, -1.2504945, -1.2004945]), ([1., 2., 3., 3., 4., 5., 6., 7., 9., 11.], [-1.7504945, -1.4404945, -1.4204945, -1.4004945, -1.3704945, -1.3204945, -1.2704945, -1.2604945, -1.2504945, -1.2004945])) for i, true_lc in enumerate(true_lcs): true_count, true_levels = true_lc lc = mm.level_crossings(kind=i+1) assert_array_almost_equal(lc.data[:10], true_count) assert_array_almost_equal(lc.args[:10], true_levels) def test_levelcrossings_extrapolate(self): tp = self.ts.turning_points() mm = tp.cycle_pairs() lc = mm.level_crossings() s = lc.sigma # x[:, 1].std() ix = slice(0, 1000, 100) lc_ray = lc.extrapolate(-2 * s, 2 * s, dist='rayleigh') assert_array_almost_equal(lc_ray.data[ix], [1.78925398e-37, 9.61028192e-23, 2.05282628e-11, 1.74389448e-03, 5.89169345e+01, 5.240000e+02, 6.72609651e+01, 4.46086175e-01, 2.23463577e-04, 8.45526153e-09]) lc_exp = lc.extrapolate(-2 * s, 2 * s, dist='expon') lc_gpd = lc.extrapolate(-2 * s, 2 * s, dist='genpareto') assert_array_almost_equal(lc_exp.data[ix], [6.51864195e-12, 1.13025876e-08, 1.95974080e-05, 3.39796881e-02, 5.89169345e+01, 5.24000000e+02, 6.43476951e+01, 1.13478843e+00, 2.00122906e-02, 3.52921977e-04]) assert_array_almost_equal(lc_gpd.data[ix], [0.00000000e+00, 0.00000000e+00, 0.00000000e+00, 0.00000000e+00, 5.89169345e+01, 5.24000000e+02, 6.80484770e+01, 1.41019390e-01, 0.00000000e+00, 0.00000000e+00]) if __name__ == "__main__": # import sys;sys.argv = ['', 'Test.testName'] unittest.main()