class TestAccuracyChecker(unittest.TestCase):
def setUp(self):
self.stats = AccuracyStats()
def test_compute_that_detected_region_is_true(self):
truth_regions = {
('url0', 0, 1): [Region('url0', 'el0', 0, 1), Region('url0', 'el1', 3, 4)],
('url1', 1, 2): [Region('url1', 'el0', 1, 2), Region('url1', 'el1', 2, 3)],
}
self.assertTrue(self.stats.is_detected_region_true(
Region('url0', 'el1', 3, 4), truth_regions))
def test_compute_that_undetected_region_is_false(self):
truth_regions = {
('url0', 0, 1): [Region('url0', 'el0', 0, 1), Region('url0', 'el1', 3, 4)],
('url1', 1, 2): [Region('url1', 'el0', 1, 2), Region('url1', 'el1', 2, 3)],
}
self.assertFalse(self.stats.is_detected_region_true(
Region('url0', 'el2', 0, 1), truth_regions))
def test_true_region_is_detected(self):
detected_regions = [Region('url0', 'el0', 0, 1), Region('url1', 'el1', 1, 2)]
region_candidates = [Region('url0', 'el0', 0, 1), Region('url0', 'el1', 1, 2)]
self.assertTrue(self.stats.is_true_region_detected(region_candidates, detected_regions))
def test_true_region_is_not_detected(self):
detected_regions = [Region('url0', 'el0', 0, 1), Region('url1', 'el1', 1, 2)]
region_candidates = [Region('url1', 'el1', 0, 1), Region('url0', 'el0', 1, 2)]
self.assertFalse(self.stats.is_true_region_detected(region_candidates, detected_regions))
def setUp(self):
self.stats = AccuracyStats()