def _biasVplotHelper(arg):
"""function to make vplot for a particular set of bed regions
"""
(chunks, params) = arg
mat = np.zeros((params.upper - params.lower, 2 * params.flank + 1))
for chunk in chunks:
try:
chunk.center()
biastrack = InsertionBiasTrack(
chunk.chrom,
chunk.start - params.flank - 1 - (params.upper / 2),
chunk.end + params.flank + params.upper / 2 + 1)
if params.bg is not None:
biastrack.read_track(params.bg, empty=0)
else:
biastrack.computeBias(params.fasta, params.chrs, params.pwm)
biasmat = BiasMat2D(chunk.chrom, chunk.start - params.flank - 1,
chunk.end + params.flank, params.lower,
params.upper)
biasmat.makeBiasMat(biastrack)
biasmat.normByInsertDist(params.fragmentsizes)
add = biasmat.get(start=chunk.start - params.flank,
end=chunk.end + params.flank,
flip=(chunk.strand == "-"))
if params.scale:
mat += add / np.sum(add)
else:
mat += add
except Exception as e:
print('Caught exception when processing:\n' + chunk.asBed() + "\n")
traceback.print_exc()
print()
raise e
return mat
class Test_BiasMat(TestCase):
"""test construction of BiasMat"""
def setUp(self):
"""setup Test_BiasMat class with construction of a biasmat"""
bed_list = ChunkList.read('example/example.bed')
self.chunk = bed_list[0]
self.biastrack = InsertionBiasTrack(self.chunk.chrom, self.chunk.start, self.chunk.end)
self.biastrack.read_track('example/example.Scores.bedgraph.gz')
self.biasmat = BiasMat2D(self.chunk.chrom,self.chunk.start+100,self.chunk.end-100,100,200)
self.biasmat.makeBiasMat(self.biastrack)
def test_biasmat1(self):
"""test case1 for biasmat"""
a1 = self.biastrack.get(pos = self.biasmat.start - 49)
a2 = self.biastrack.get(pos = self.biasmat.start + 50)
correct = np.exp(a1+a2)
self.assertTrue(abs(correct - self.biasmat.mat[0,0])<0.01*correct)
def test_biasmat2(self):
"""test case2 for biasmat"""
a1 = self.biastrack.get(pos = self.biasmat.start + 145)
a2 = self.biastrack.get(pos = self.biasmat.start + 295)
correct = np.exp(a1+a2)
self.assertTrue(abs(correct - self.biasmat.mat[51,220]) < 0.01*correct)
def test_normByInsertDist(self):
"""test that normalization by insert distribution works as expected"""
isizes = FragmentSizes(lower=100,upper=200, vals = np.array(range(100,200)))
self.biasmat.normByInsertDist(isizes)
a1 = self.biastrack.get(pos = self.biasmat.start -50)
a2 = self.biastrack.get(pos = self.biasmat.start + 50)
correct = np.exp(a1+a2)*isizes.get(size = 101)
self.assertTrue(abs(correct - self.biasmat.mat[1,0])<0.01*correct)
def _biasVplotHelper(arg):
"""function to make vplot for a particular set of bed regions
"""
(chunks, params) = arg
mat = np.zeros((params.upper-params.lower,2*params.flank+1))
for chunk in chunks:
try:
chunk.center()
biastrack = InsertionBiasTrack(chunk.chrom, chunk.start - params.flank - 1 - (params.upper/2),
chunk.end + params.flank + params.upper/2+1)
if params.bg is not None:
biastrack.read_track(params.bg, empty = 0)
else:
biastrack.computeBias(params.fasta, params.chrs, params.pwm)
biasmat = BiasMat2D(chunk.chrom, chunk.start - params.flank - 1, chunk.end + params.flank,
params.lower, params.upper)
biasmat.makeBiasMat(biastrack)
biasmat.normByInsertDist(params.fragmentsizes)
add = biasmat.get(start = chunk.start - params.flank, end = chunk.end + params.flank,
flip = (chunk.strand == "-"))
if params.scale:
mat += add/np.sum(add)
else:
mat += add
except Exception as e:
print('Caught exception when processing:\n' + chunk.asBed() + "\n")
traceback.print_exc()
print()
raise e
return mat
def setUp(self):
""" set up class for testing variance calculation for background signal
"""
bed_list = ChunkList.read('example/example.bed')
chunk = bed_list[0]
vmat = V.VMat.open('example/example.VMat')
biastrack = InsertionBiasTrack(chunk.chrom, chunk.start, chunk.end)
biastrack.read_track('example/example.Scores.bedgraph.gz')
biasmat = BiasMat2D(chunk.chrom,chunk.start+200,chunk.end-200,100,250)
biasmat.makeBiasMat(biastrack)
self.signaldist = Nuc.SignalDistribution(chunk.start+300,vmat,biasmat,35)
def setUp(self):
""" set up class for testing variance calculation for background signal
"""
bed_list = ChunkList.read('example/example.bed')
chunk = bed_list[0]
vmat = V.VMat.open('example/example.VMat')
biastrack = InsertionBiasTrack(chunk.chrom, chunk.start, chunk.end)
biastrack.read_track('example/example.Scores.bedgraph.gz')
biasmat = BiasMat2D(chunk.chrom,chunk.start+200,chunk.end-200,100,250)
biasmat.makeBiasMat(biastrack)
self.signaldist = Nuc.SignalDistribution(chunk.start+300,vmat,biasmat,35)
class Test_BiasMat(TestCase):
"""test construction of BiasMat"""
def setUp(self):
"""setup Test_BiasMat class with construction of a biasmat"""
bed_list = ChunkList.read('example/example.bed')
self.chunk = bed_list[0]
self.biastrack = InsertionBiasTrack(self.chunk.chrom, self.chunk.start,
self.chunk.end)
self.biastrack.read_track('example/example.Scores.bedgraph.gz')
self.biasmat = BiasMat2D(self.chunk.chrom, self.chunk.start + 100,
self.chunk.end - 100, 100, 200)
self.biasmat.makeBiasMat(self.biastrack)
def test_biasmat1(self):
"""test case1 for biasmat"""
a1 = self.biastrack.get(pos=self.biasmat.start - 49)
a2 = self.biastrack.get(pos=self.biasmat.start + 50)
correct = np.exp(a1 + a2)
self.assertTrue(abs(correct - self.biasmat.mat[0, 0]) < 0.01 * correct)
def test_biasmat2(self):
"""test case2 for biasmat"""
a1 = self.biastrack.get(pos=self.biasmat.start + 145)
a2 = self.biastrack.get(pos=self.biasmat.start + 295)
correct = np.exp(a1 + a2)
self.assertTrue(
abs(correct - self.biasmat.mat[51, 220]) < 0.01 * correct)
def test_normByInsertDist(self):
"""test that normalization by insert distribution works as expected"""
isizes = FragmentSizes(lower=100,
upper=200,
vals=np.array(range(100, 200)))
self.biasmat.normByInsertDist(isizes)
a1 = self.biastrack.get(pos=self.biasmat.start - 50)
a2 = self.biastrack.get(pos=self.biasmat.start + 50)
correct = np.exp(a1 + a2) * isizes.get(size=101)
self.assertTrue(abs(correct - self.biasmat.mat[1, 0]) < 0.01 * correct)
