Пример #1
0
    def test_cover_r(self):
        self.args.tumor_files = ['test/test_normal_1.txt']
        self.args.non_tumor_files = ['test/test_normal_1.txt']
        self.args.r = 1.5
        patient = Patient(self.args, 0)

        # mean is 2 for C1.
        self.assertTrue(patient.covers(3.01, 'C1'))
        self.assertTrue(patient.covers(4, 'C1'))
        self.assertTrue(patient.covers(100, 'C1'))
        self.assertFalse(patient.covers(0, 'C1'))
        self.assertFalse(patient.covers(1, 'C1'))
        self.assertFalse(patient.covers(2.5, 'C1'))
        self.assertFalse(patient.covers(3, 'C1'))

        # mean is 1.25 for C2.
        self.assertTrue(patient.covers(1.25 * 1.5 + 0.01, 'C2'))
        self.assertTrue(patient.covers(4, 'C2'))
        self.assertTrue(patient.covers(100, 'C2'))
        self.assertFalse(patient.covers(0, 'C2'))
        self.assertFalse(patient.covers(1, 'C2'))
        self.assertFalse(patient.covers(1.25, 'C2'))

        # mean is 3 for C3.
        self.assertTrue(patient.covers(4.6, 'C3'))
        self.assertTrue(patient.covers(5, 'C3'))
        self.assertTrue(patient.covers(100, 'C3'))
        self.assertFalse(patient.covers(0, 'C3'))
        self.assertFalse(patient.covers(1, 'C3'))
        self.assertFalse(patient.covers(1.25, 'C3'))
        self.assertFalse(patient.covers(3, 'C3'))
        self.assertFalse(patient.covers(4.5, 'C3'))