def test_calch2(self):
logging.info("Leap test_calch2")
related_file = os.path.abspath(
os.path.join(self.gold_dir, 'dataset1.related'))
bed, _ = leapUtils.loadData(self.bedbase,
None,
self.phen_fn,
loadSNPs=False)
keepArr = leapUtils.loadRelatedFile(bed, related_file)
for i in range(1, 11):
h2_file = os.path.abspath(
os.path.join(self.gold_dir, 'dataset1_nochr{0}.h2'.format(i)))
eigen_file = os.path.abspath(
os.path.join(self.tempout_dir,
'dataset1_nochr{}.npz'.format(i)))
eigen = np.load(eigen_file)
extractSim = 'dataset1/extracts/nochr{0}_extract.txt'.format(i)
expected_h2 = np.loadtxt(h2_file, usecols=[0])
h2 = leapMain.calcH2(self.phen_fn,
0.001,
eigen,
keepArr=keepArr,
h2coeff=1.0)
assert np.abs(
h2 - expected_h2) < 1e-5, 'Incorrect heritability estimated'
def test_calch2(self):
logging.info("Leap test_calch2")
related_file = os.path.abspath(os.path.join(self.gold_dir, 'dataset1.related'))
bed, _ = leapUtils.loadData(self.bedbase, None, self.phen_fn, loadSNPs=False)
keepArr = leapUtils.loadRelatedFile(bed, related_file)
for i in xrange(1,11):
h2_file = os.path.abspath(os.path.join(self.gold_dir, 'dataset1_nochr{0}.h2'.format(i)))
eigen_file = os.path.abspath(os.path.join(self.tempout_dir, 'dataset1_nochr{}.npz'.format(i)))
eigen = np.load(eigen_file)
extractSim = 'dataset1/extracts/nochr{0}_extract.txt'.format(i)
expected_h2 = np.loadtxt(h2_file, usecols=[0])
h2 = leapMain.calcH2(self.phen_fn, 0.001, eigen, keepArr=keepArr, h2coeff=1.0)
assert np.abs(h2-expected_h2)<1e-5, 'Incorrect heritability estimated'
def test_probit(self):
logging.info("Leap test_probit")
related_file = os.path.abspath(os.path.join(self.gold_dir, 'dataset1.related'))
bed, _ = leapUtils.loadData(self.bedbase, None, self.phen_fn, loadSNPs=False)
keepArr = leapUtils.loadRelatedFile(bed, related_file)
for i in xrange(1,11):
h2_file = os.path.abspath(os.path.join(self.gold_dir, 'dataset1_nochr{0}.h2'.format(i)))
h2 = np.loadtxt(h2_file, usecols=[0])
eigen_file = os.path.abspath(os.path.join(self.tempout_dir, 'dataset1_nochr{}.npz'.format(i)))
eigen = np.load(eigen_file)
ref_file = os.path.abspath(os.path.join(self.gold_dir, 'dataset1_nochr{}.liabs'.format(i)))
extractSim = 'dataset1/extracts/nochr{0}_extract.txt'.format(i)
bed, _ = leapUtils.loadData(self.bedbase, extractSim, self.phen_fn, loadSNPs=True)
output_file = os.path.abspath(os.path.join(self.tempout_dir, 'dataset1_nochr{}'.format(i)))
leapMain.probit(bed, self.phen_fn, h2=h2, prev=0.001, eigen=eigen, outFile=output_file, keepArr=keepArr, hess=False)
self.compare_pheno(ref_file, output_file+'.liabs')
