Mercurial > repos > yufei-luo > s_mart
view commons/core/coord/test/Test_SlidingWindow.py @ 14:c79b9ae3f65f
Deleted selected files
author | m-zytnicki |
---|---|
date | Fri, 19 Apr 2013 10:13:11 -0400 |
parents | 769e306b7933 |
children |
line wrap: on
line source
import unittest from commons.core.coord.SlidingWindow import SlidingWindow from commons.core.coord.SlidingWindow import SlidingWindowToCountMatchingBases from commons.core.coord.Set import Set class Test_SlidingWindow( unittest.TestCase ): def test_slideWindowOnce( self ): expStart = 91 expEnd = 190 self.sw = SlidingWindow(100, 10) self.sw.slideWindowOnce() obsStart = self.sw._start obsEnd = self.sw._end self.assertEqual(expStart, obsStart) self.assertEqual(expEnd, obsEnd) def test_slideWindowOnceFourTime( self ): expStart = 201 expEnd = 300 self.sw = SlidingWindow(100, 50) i = 0 for i in range(4): self.sw.slideWindowOnce() i += 1 obsStart = self.sw._start obsEnd = self.sw._end self.assertEqual(expStart, obsStart) self.assertEqual(expEnd, obsEnd) class Test_SlidingWindowToCountMatchingBases(unittest.TestCase): def test_getSetLengthOnWindow_featureIncluded( self ): self.sw = SlidingWindowToCountMatchingBases(100, 1) iSet = Set( 1, "TE3", "chr1", 21, 30 ) exp = 10 obs = self.sw.getSetLengthOnWindow( iSet) self.assertEqual( exp, obs ) def test_getSetLengthOnWindow_windowIncluded( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 21, 530 ) exp = 100 obs = self.sw.getSetLengthOnWindow( iSet) self.assertEqual( exp, obs ) def test_getSetLengthOnWindow_featureOverlapLeft( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 21, 130 ) exp = 40 obs = self.sw.getSetLengthOnWindow( iSet) self.assertEqual( exp, obs ) def test_getSetLengthOnWindow_featureOverlapRight( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 121, 230 ) exp = 70 obs = self.sw.getSetLengthOnWindow( iSet) self.assertEqual( exp, obs ) def test_getCoordSetOnWindow_featureIncluded( self ): self.sw = SlidingWindowToCountMatchingBases(100, 1) iSet = Set( 1, "TE3", "chr1", 21, 30 ) expStart = 21 expEnd = 30 obsStart,obsEnd = self.sw.getCoordSetOnWindow( iSet) self.assertEqual( expStart, obsStart ) self.assertEqual( expEnd, obsEnd ) def test_getCoordSetOnWindow_windowIncluded( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 21, 530 ) expStart = 91 expEnd = 190 obsStart,obsEnd = self.sw.getCoordSetOnWindow( iSet) self.assertEqual( expStart, obsStart ) self.assertEqual( expEnd, obsEnd ) def test_getCoordSetOnWindow_featureOverlapLeft( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 21, 130 ) expStart = 91 expEnd = 130 obsStart,obsEnd = self.sw.getCoordSetOnWindow( iSet) self.assertEqual( expStart, obsStart ) self.assertEqual( expEnd, obsEnd ) def test_getCoordSetOnWindow_featureOverlapRight( self ): self.sw = SlidingWindowToCountMatchingBases(100, 10) self.sw.slideWindowOnce() iSet = Set( 1, "TE3", "chr1", 121, 230 ) expStart = 121 expEnd = 190 obsStart,obsEnd = self.sw.getCoordSetOnWindow( iSet) self.assertEqual( expStart, obsStart ) self.assertEqual( expEnd, obsEnd ) test_suite = unittest.TestSuite() test_suite.addTest( unittest.makeSuite( Test_SlidingWindow ) ) if __name__ == "__main__": unittest.TextTestRunner(verbosity=2).run( test_suite )