def test_pca_mssa_ec(self):
data = setup_data2(nx=30, ny=20)
span = Analyzer(data)
span.mssa(nmssa=4)
ec = span.mssa_ec(xeof=span.mssa_eof(), xdata=data)
pc = span.mssa_pc()
self.assertTrue(npy.allclose(ec, pc))
def test_mssa_eof(self):
data = setup_data2(nx=3, ny=2)
span = Analyzer(data)
eof = span.mssa_eof(nmssa=4)
self.assertTrue(npy.allclose(
eof[1,1].compressed(),
npy.array([ 0.10764546, -0.0763087 , -0.07211439, 0.05060925, 0.128224 ])))
def test_pca_mssa_rec(self):
data = setup_data2(nx=30, ny=20)
span = Analyzer(data)
rec = span.mssa_rec(nmssa=10)
self.assertTrue(npy.allclose(
rec[2,:2,:2].compressed(),
npy.array([ 0.36712853, 0.28310489, 0.27881545])))
def test_mssa_eof_restack(self):
data = setup_data2(nx=3, ny=2)
span = Analyzer(data)
steof = span.mssa_eof(nmssa=4)
stacked_eof = npy.ascontiguousarray(span.restack(steof, scale=False))
self.assertTrue(npy.allclose(steof[1,1].compressed(), stacked_eof[:,1,1]))
raw_eof = span._mssa_raw_eof.reshape(stacked_eof.shape)
self.assertTrue(npy.allclose(stacked_eof, raw_eof))
def test_pca_mssa_xrec(self):
data = setup_data2(nx=30, ny=20)
span = Analyzer(data)
rec = span.mssa_rec(nmssa=4)
xeof = span.mssa_eof()
xpc = span.mssa_pc()
xrec = span.mssa_rec(xeof=xeof, xpc=xpc)
self.assertTrue(npy.allclose(rec, xrec))
def test_pca_mssa(self):
data = setup_data2(nx=30, ny=20)
span = Analyzer(data, nmssa=2)
self.assertTrue(npy.allclose(
span.mssa_ev(),
npy.array([ 2482.65882191, 2355.53413819])))
pc = span.mssa_pc()
self.assertTrue(npy.allclose(
span.mssa_pc()[:,:2].ravel(),
npy.array([-19.47701848, -20.97653195, 2.99480916, -0.20444392])))
self.assertAlmostEqual(span.mssa_eof()[1,2,3,4], 0.0041007559488027926)
def test_mssa_rec(self):
data = setup_data2(nx=3, ny=2, xyfact=0)
span = Analyzer(data)
rec = span.mssa_rec(nmssa=20)
print span.mssa_ev(cumsum=True, relative=True)
# import pylab as P;P.subplot(211);P.pcolormesh(data.reshape(-1, 6));P.subplot(212);P.pcolormesh(rec.reshape(-1, 6));P.show()
print rec[50].compressed()
print data[50].compressed()
print (rec-data).var()/data.var()
self.assertTrue(npy.allclose(
rec[50].compressed(),
npy.array([0.30848007, 0.8622368, 0.96028445, 0.8236466, 0.56329503 ])))
def test_mssa(self):
data = setup_data2(nx=3, ny=2)
span = Analyzer(data)
span.mssa(nmssa=4)
print data.reshape(120, 6).mask
# self.assertTrue(npy.allclose(
# span._mssa_raw_ev,
# npy.array([ 84.65284564, 39.94531834, 36.70936929, 21.49023578])))
print span._mssa_raw_pc[:2,:2]
self.assertTrue(npy.allclose(
span._mssa_raw_pc[:2,:2],
npy.array([[-13.66377591, 2.62887011],
[-13.28622293, 2.74443558]])))
self.assertTrue(npy.allclose(
span._mssa_raw_eof[:2,:2],
npy.array([[ 0.05067246, 0.10596479],
[ 0.05300994, 0.10764546]])))
def test_pca_ndim3(self):
A = Analyzer(setup_data2())
A.pca()
self.assertTrue(npy.allclose(
npy.abs(A.pca_eof(raw=True)[190:193,5]),
npy.abs(npy.array([-0.03623882, -0.02312414, -0.00942417]))))