def test_pickle():
import tempfile
from numpy.testing import assert_equal
tmpdir = tempfile.mkdtemp(prefix='pickle')
a = range(10)
save_pickle(a, tmpdir+'/res.pkl')
b = load_pickle(tmpdir+'/res.pkl')
assert_equal(a, b)
#cleanup, tested on Windows
try:
import os
os.remove(tmpdir+'/res.pkl')
os.rmdir(tmpdir)
except (OSError, IOError):
pass
assert not os.path.exists(tmpdir)
#test with file handle
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO()
save_pickle(a, fh)
fh.seek(0,0)
c = load_pickle(fh)
fh.close()
assert_equal(a,b)
def test_pickle(self):
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO()
#test wrapped results load save pickle
self.res.save(fh)
fh.seek(0,0)
res_unpickled = self.res.__class__.load(fh)
assert_(type(res_unpickled) is type(self.res))
def check_pickle(obj):
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO()
pickle.dump(obj, fh)
plen = fh.tell()
fh.seek(0, 0)
res = pickle.load(fh)
fh.close()
return res, plen
def test_pickle_wrapper(self):
from gwstatsmodels.iolib.smpickle import save_pickle, load_pickle
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO() #use cPickle with binary content
#test unwrapped results load save pickle
self.results._results.save(fh)
fh.seek(0, 0)
res_unpickled = self.results._results.__class__.load(fh)
assert_(type(res_unpickled) is type(self.results._results))
#test wrapped results load save
fh.seek(0, 0)
#save_pickle(self.results, fh)
self.results.save(fh)
fh.seek(0, 0)
#res_unpickled = load_pickle(fh)
res_unpickled = self.results.__class__.load(fh)
fh.close()
#print type(res_unpickled)
assert_(type(res_unpickled) is type(self.results))
before = sorted(self.results.__dict__.keys())
after = sorted(res_unpickled.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results._results.__dict__.keys())
after = sorted(res_unpickled._results.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results.model.__dict__.keys())
after = sorted(res_unpickled.model.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results._cache.keys())
after = sorted(res_unpickled._cache.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
def test_pickle():
import tempfile
from numpy.testing import assert_equal
tmpdir = tempfile.mkdtemp(prefix='pickle')
a = range(10)
save_pickle(a, tmpdir + '/res.pkl')
b = load_pickle(tmpdir + '/res.pkl')
assert_equal(a, b)
#cleanup, tested on Windows
try:
import os
os.remove(tmpdir + '/res.pkl')
os.rmdir(tmpdir)
except (OSError, IOError):
pass
assert not os.path.exists(tmpdir)
#test with file handle
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO()
save_pickle(a, fh)
fh.seek(0, 0)
c = load_pickle(fh)
fh.close()
assert_equal(a, b)
def check_pickle(obj):
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO()
pickle.dump(obj, fh)
plen = fh.tell()
fh.seek(0,0)
res = pickle.load(fh)
fh.close()
return res, plen
def test_pickle_wrapper(self):
from gwstatsmodels.iolib.smpickle import save_pickle, load_pickle
from gwstatsmodels.compatnp.py3k import BytesIO
fh = BytesIO() #use cPickle with binary content
#test unwrapped results load save pickle
self.results._results.save(fh)
fh.seek(0,0)
res_unpickled = self.results._results.__class__.load(fh)
assert_(type(res_unpickled) is type(self.results._results))
#test wrapped results load save
fh.seek(0,0)
#save_pickle(self.results, fh)
self.results.save(fh)
fh.seek(0,0)
#res_unpickled = load_pickle(fh)
res_unpickled = self.results.__class__.load(fh)
fh.close()
#print type(res_unpickled)
assert_(type(res_unpickled) is type(self.results))
before = sorted(self.results.__dict__.keys())
after = sorted(res_unpickled.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results._results.__dict__.keys())
after = sorted(res_unpickled._results.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results.model.__dict__.keys())
after = sorted(res_unpickled.model.__dict__.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
before = sorted(self.results._cache.keys())
after = sorted(res_unpickled._cache.keys())
assert_(before == after, msg='not equal %r and %r' % (before, after))
ss5 = '''\
2 - 3 4.340 0.691 7.989 ***
2 - 1 4.600 0.951 8.249 ***
3 - 2 -4.340 -7.989 -0.691 ***
3 - 1 0.260 -3.389 3.909 -
1 - 2 -4.600 -8.249 -0.951 ***
1 - 3 -0.260 -3.909 3.389 '''
#accommodate recfromtxt for python 3.2, requires bytes
ss = asbytes(ss)
ss2 = asbytes(ss2)
ss3 = asbytes(ss3)
ss5 = asbytes(ss5)
dta = np.recfromtxt(BytesIO(ss), names=("Rust", "Brand", "Replication"))
dta2 = np.recfromtxt(BytesIO(ss2),
names=("idx", "Treatment", "StressReduction"))
dta3 = np.recfromtxt(BytesIO(ss3), names=("Brand", "Relief"))
dta5 = np.recfromtxt(BytesIO(ss5),
names=('pair', 'mean', 'lower', 'upper', 'sig'),
delimiter='\t')
sas_ = dta5[[1, 3, 2]]
from gwstatsmodels.stats.multicomp import (tukeyhsd, pairwise_tukeyhsd,
MultiComparison)
#import gwstatsmodels.sandbox.stats.multicomp as multi
#print tukeyhsd(dta['Brand'], dta['Rust'])
def get_thsd(mci, alpha=0.05):
