def secondTests():
writer= uWriter("","BEDGRAPH")
bedParser = uParser("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED")
A = uBasicExp()
A.load_with_parser(bedParser,0)
A.write_with_writer(writer)
C = uBasicExp()
bedParser = uParser("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED")
C.load_with_parser(bedParser,0)
C.removeChr("chr2")
C.write_with_writer
print "before anything"
A.write_with_writer(writer)
L=A.get_overlapping_from_exp(C)
print "after get overlapping exp"
L.write_with_writer(writer)
L=A.get_overlapping_from_region("chr20",257180,257480)
print "after get overlapping region"
L.write_with_writer(writer)
#Need to test
#get_overlapping_from_chrom(self,chrom):
print "reset"
bedParser = uParser("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED")
A = uBasicExp()
A.load_with_parser(bedParser,0)
A.write_with_writer(writer)
print "after removeDistinct"
A.sort_sites()
A.remove_distinct("chr20",250000, 257700)
A.write_with_writer(writer)
print "reset"
bedParser = uParser("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED")
A = uBasicExp()
A.load_with_parser(bedParser,0)
print "after getDistinct"
A.sort_sites()
F=A.get_distinct("chr20",250000, 257700)
print F.count()
F.write_with_writer(writer)
print "ChrCount", F.get_chr_count()
print "Get subset no chrom"
print A.get_subset_count("chr22",250000, 257700)
deleteA= uBasic()
def ManyTests():
print "--------------------------"
print "TESTS: uBasicNGSChrom"
print "--------------------------"
print "**** Tests for constructor"
print "No param:"
chrom = uBasicChrom()
print "chr: " + chrom.get_chr()
print "chrSize: " + str(chrom.get_chrom_size())
print "Chrom Name"
chromName = uBasicChrom("chr5")
print "chr: " + chromName.get_chr()
print "chrSize: " + str(chromName.get_chrom_size())
print "Chrom Name Size:"
chromNameSize = uBasicChrom("chr3", 123456)
print "chr: " + chromNameSize.get_chr()
print "chrSize: " + str(chromNameSize.get_chrom_size())
print "Chrom Size:"
chrom_size = uBasicChrom(chromosomeSize=123456)
print "chr: " + chrom_size.get_chr()
print "chrSize: " + str(chrom_size.get_chrom_size())
print ""
print "**** Tests for get_copy"
chromCopy = chromNameSize.get_copy()
print "chr: " + chromCopy.get_chr()
print "chrSize: " + str(chromCopy.get_chrom_size())
print ""
print "**** Tests for add_data functions"
chromAddData = uBasicChrom("chr3")
basicAddData = uBasic("chr3", 100, 200)
chromAddData.add_data(basicAddData)
print "chr: " + chromAddData.get_chr()
print "chrSize: " + str(chromAddData.get_chrom_size())
print ""
print "**** Tests for infer_chr_size "
chromInfer = uBasicChrom("chr3")
chromInfer.add_data(uBasic("chr3", 1000, 3000))
print "chrSize (before): " + str(chromInfer.get_chrom_size())
chromInfer.infer_chr_size()
print "chrSize (after): " + str(chromInfer.get_chrom_size())
chromInfer = uBasicChrom()
print "chrSize (empty):" + str(chromInfer.get_chrom_size())
print ""
print "**** Tests for the divide_items_into_n_bins function"
chromDivide = uBasicChrom("chr4")
chromDivide.add_data(uBasic("chr4", 1000, 3000))
print "count (before): " + str(chromDivide.count())
chromDivide.divide_items_into_n_bins(10, "IGNORE")
print "count (after): " + str(chromDivide.count())
print ""
print "**** Tests for the divideItemsIntoBinofSize function"
chromDivide = uBasicChrom("chr4")
chromDivide.add_data(uBasic("chr4", 1000, 2000))
print "count (before): " + str(chromDivide.count())
chromDivide.divide_items_into_bin_of_size(20, "IGNORE")
print "count (after): " + str(chromDivide.count())
print ""
print "**** Tests for the function getSite"
chromGetSite = uBasicChrom("chr1")
chromGetSite.add_data(uBasic("chr1", 300, 500))
aSite = chromGetSite.get_site(0)
print "getSite: chr: " + aSite.get_chr()
print "getSite: start: " + str(aSite.get_start())
print "getSite: end: " + str(aSite.get_end())
print ""
print "**** Tests for the statistic functions"
chromStats = uBasicChrom("chrX")
chromStats.add_data(uBasic("chrX", 300, 500))
chromStats.add_data(uBasic("chrX", 300, 600))
chromStats.add_data(uBasic("chrX", 400, 500))
print "avg: " + str(chromStats.avg_site_size())
print "min: " + str(chromStats.min_site_size())
print "max: " + str(chromStats.max_site_size())
print "sum: " + str(chromStats.sum_site_size())
print "count: " + str(chromStats.count())
print "**** Tests for adding random sites"
chromAddRandom = uBasicChrom("chrY")
chromAddRandom.set_chr_size(100000)
chromAddRandom.add_random_sites(100, 10,10)
print "Count: " + str(chromAddRandom.count())
writer= uWriter("","BEDGRAPH")
chromAddRandom.write_with_writer(writer)
print ""
print "**** Tests for get_overlapping/get_overlapping_count (other Chrom)"
chromOverlap1 = uBasicChrom("chr1")
chromOverlap1.add_data(uBasic("chr1", 100, 200))
chromOverlap1.add_data(uBasic("chr1", 400, 800))
chromOverlap1.add_data(uBasic("chr1", 1000, 1200))
print "Different chrom:"
chromOverlap2 = uBasicChrom("chr2")
chromOverlap2.add_data(uBasic("chr2", 100, 200))
chromOverlap2.add_data(uBasic("chr2", 400, 800))
chromOverlap2.add_data(uBasic("chr2", 1000, 1200))
chromOverlapResult = chromOverlap1.get_overlapping(chromOverlap2)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromOverlap1.get_overlapping_count(chromOverlap2))
print "No overlap:"
chromOverlap3 = uBasicChrom("chr1")
chromOverlap3.add_data(uBasic("chr1", 1300, 1400))
chromOverlap3.add_data(uBasic("chr1", 1400, 1800))
chromOverlap3.add_data(uBasic("chr1", 11000, 11200))
chromOverlapResult = chromOverlap1.get_overlapping(chromOverlap3)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromOverlap1.get_overlapping_count(chromOverlap3))
print "Some overlap:"
chromOverlap4 = uBasicChrom("chr1")
chromOverlap4.add_data(uBasic("chr1", 100, 150))
chromOverlap4.add_data(uBasic("chr1", 400, 900))
chromOverlap4.add_data(uBasic("chr1", 11000, 11200))
chromOverlapResult = chromOverlap1.get_overlapping(chromOverlap4)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromOverlap1.get_overlapping_count(chromOverlap4))
print "All overlap: "
chromOverlap5 = uBasicChrom("chr1")
chromOverlap5.add_data(uBasic("chr1", 100, 200))
chromOverlap5.add_data(uBasic("chr1", 400, 800))
chromOverlap5.add_data(uBasic("chr1", 1000, 1200))
chromOverlapResult = chromOverlap1.get_overlapping(chromOverlap5)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromOverlap1.get_overlapping_count(chromOverlap5))
print ""
print "**** Tests for get_overlapping/get_overlapping_count (basic)"
chromBasicOverlap1 = uBasicChrom("chr2")
chromBasicOverlap1.add_data(uBasic("chr2", 100, 200))
chromBasicOverlap1.add_data(uBasic("chr2", 400, 800))
chromBasicOverlap1.add_data(uBasic("chr2", 1000, 1200))
print "chromBasicOverlap1.count(): " + str(chromBasicOverlap1.count())
basicOverlap = uBasic("chr2", 50, 900)
chromOverlapResult = chromBasicOverlap1.get_overlapping_basic(basicOverlap)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromBasicOverlap1.get_overlapping_count_basic(basicOverlap))
aSite = chromOverlapResult.get_site(0)
print "get_site: chr: " + aSite.get_chr()
print "get_site: start: " + str(aSite.get_start())
print "get_site: end: " + str(aSite.get_end())
print ""
print "**** Tests for get_overlapping/get_overlapping_count (region)"
chromRegionOverlap1 = uBasicChrom("chr3")
chromRegionOverlap1.add_data(uBasic("chr3", 100, 200))
chromRegionOverlap1.add_data(uBasic("chr3", 400, 800))
chromRegionOverlap1.add_data(uBasic("chr3", 1000, 1200))
chromOverlapResult = chromRegionOverlap1.get_overlapping_region(50, 900)
print "Result Count: " + str(chromOverlapResult.count())
print "get_overlapCount: " + str(chromBasicOverlap1.get_overlapping_count_region(50, 900))
print ""
print "**** Tests for get_not_overlapping (other Chrom)"
chromNotOverlap1 = uBasicChrom("chr1")
chromNotOverlap1.add_data(uBasic("chr1", 100, 200))
chromNotOverlap1.add_data(uBasic("chr1", 400, 800))
chromNotOverlap1.add_data(uBasic("chr1", 1000, 1200))
print "Different chrom:"
chromNotOverlap2 = uBasicChrom("chr2")
chromNotOverlap2.add_data(uBasic("chr2", 100, 200))
chromNotOverlap2.add_data(uBasic("chr2", 400, 800))
chromNotOverlap2.add_data(uBasic("chr2", 1000, 1200))
chromNotOverlapResult = chromNotOverlap1.get_not_overlapping(chromNotOverlap2)
print "Result Count: " + str(chromNotOverlapResult.count())
# print "get_not_overlapCount: " + str(chromNotOverlap1.get_not_overlapping_count(chromNotOverlap2))
print "No overlap:"
chromNotOverlap3 = uBasicChrom("chr1")
chromNotOverlap3.add_data(uBasic("chr1", 1300, 1400))
chromNotOverlap3.add_data(uBasic("chr1", 1400, 1800))
chromNotOverlap3.add_data(uBasic("chr1", 11000, 11200))
chromNotOverlapResult = chromNotOverlap1.get_not_overlapping(chromNotOverlap3)
print "Result Count: " + str(chromNotOverlapResult.count())
# print "get_not_overlapCount: " + str(chromNotOverlap1.get_not_overlapping_count(chromNotOverlap3))
print "Some overlap:"
chromNotOverlap4 = uBasicChrom("chr1")
chromNotOverlap4.add_data(uBasic("chr1", 100, 150))
chromNotOverlap4.add_data(uBasic("chr1", 400, 900))
chromNotOverlap4.add_data(uBasic("chr1", 11000, 11200))
chromNotOverlapResult = chromNotOverlap1.get_not_overlapping(chromNotOverlap4)
print "Result Count: " + str(chromNotOverlapResult.count())
# print "get_overlapCount: " + str(chromNotOverlap1.get_not_overlapping_count(chromNotOverlap4))
print "All overlap: "
chromNotOverlap5 = uBasicChrom("chr1")
chromNotOverlap5.add_data(uBasic("chr1", 100, 200))
chromNotOverlap5.add_data(uBasic("chr1", 400, 800))
chromNotOverlap5.add_data(uBasic("chr1", 1000, 1200))
chromNotOverlapResult = chromNotOverlap1.get_not_overlapping(chromNotOverlap5)
print "Result Count: " + str(chromNotOverlapResult.count())
def firstTests():
from uBasicNGS_wrapper import uBasic
from uParser_wrapper import uParser
from uWriter_Wrapper import uWriter
bedParser = uParser("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED")
print "Make empty BasicExp"
A = uBasicExp()
A.sort_sites()
B= uBasic("chr2",100,200)
print "Test is Chrom"
if (A.is_chrom("chr1")==0):
print "Chrom check ok"
else:
print A.is_chrom("chr1")
print "Test count"
print A.count()
print "Test add Data"
A.add_data_unit(B)
A.add_data_unit(uBasic("chr2",50,100))
print A.count()
if (A.is_chrom("chr2")==1):
print "Chrom check2 ok"
if (A.is_chrom("chr1")==1):
print "Chrom check fail"
print "before sort"
print A.is_sorted()
print "Test Sort"
A.sort_sites()
print "after sort"
print A.is_sorted()
print "Test Load With Parser"
A.load_with_parser(bedParser,5)
print A.count()
A.load_with_parser_path("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED",3)
print A.count()
writer= uWriter("","BEDGRAPH")
B= A.get_site("chr20",2)
print B.get_chr(),B.get_start(),B.get_end()
print A.count()
A.removeChr("chr20")
print A.count()
A.write_with_writer(writer)
A.sort_sites()
C = A.find_preceding_site("chr2",0)
print C
C = A.find_next_site("chr2",600)
print C
print "----"
A.remove_site("chr2",C)
A.write_with_writer(writer)
print "reloading"
A.load_with_parser_path("/home/local/USHERBROOKE/nora2001/Work/class/NGS_testing/data/BED/bedH2AZ.bed","BED",0)
print "before subset"
A.sort_sites()
A.write_with_writer(writer)
#F = BasicExp()
A.remove_subset("chr2",30,120)
print "after subset"
A.write_with_writer(writer)
print "before setSize"
print A.get_chr_size("chr2")
print "after setSize"
A.set_chr_size("chr2",22999)
print A.get_chr_size("chr2")
print "after infer"
A.infer_chr_size()
print A.get_chr_size("chr2")
