def test_finding_closest_match(self):
auto = EPGProfile.from_file(example_file('simple-auto.csv'))
manual = EPGProfile.from_file(example_file('simple-manual.csv'))
auto, manual = _EPGFingerprint(auto), _EPGFingerprint(manual)
finder = auto.find_closest_match_to
self.assertEqual((1941, 3000), finder(manual, 1947, 3000))
self.assertEqual((235, 560), finder(manual, 215, 550))
self.assertEqual((4016, 4638), finder(manual, 4014, 4638))
def test_calculate_offset(self):
auto = EPGProfile.from_file(example_file('simple-auto.csv'))
manual = EPGProfile.from_file(example_file('simple-manual.csv'))
auto, manual = _EPGFingerprint(auto), _EPGFingerprint(manual)
expected = dict(
pd=[[20], [10]],
c=[[10, 0], [-446, 2]],
np=[[-6, 2], [0, 0]])
offsets = EPGComparator._calculate_offsets((auto, manual),
fractional=False)
self.assertDictEqual(expected, offsets)
def test_making_fingerprint(self):
profile = EPGProfile.from_file(example_file('simple-manual.csv'))
expected = [None] * 215
expected.extend(['pd'] * 336)
expected.extend(['c'] * 1396)
expected.extend(['np'] * 1054)
expected.extend(['c'] * 1013)
expected.extend(['np'] * 625)
self.assertSequenceEqual(expected, _EPGFingerprint(profile))
def test_finding_closest_match_complex(self):
auto = EPGProfile.from_file(example_file('Ap1-haba-auto.csv'))
manual = EPGProfile.from_file(example_file('Ap1-haba-manual.csv'))
auto, manual = _EPGFingerprint(auto), _EPGFingerprint(manual)
finder = auto.find_closest_match_to
self.assertEqual((0, 19313), finder(manual, 0, 19313))
self.assertEqual(None, finder(manual, 19663, 66200))
self.assertEqual((66201, 98635), finder(manual, 66201, 98635))
self.assertEqual(None, finder(manual, 101423, 101845))
self.assertEqual(None, finder(manual, 203058, 209251))
self.assertEqual(None, finder(manual, 209667, 214063))
self.assertEqual(None, finder(manual, 214507, 234165))
self.assertEqual((249608, 251075), finder(manual, 249608, 251441))
self.assertEqual(None, finder(manual, 251442, 359998))
self.assertEqual(None, finder(manual, 371844, 382845))
self.assertEqual(None, finder(manual, 382985, 389107))
self.assertEqual((389108, 389470), finder(manual, 389108, 389470))
self.assertEqual((424665, 425297), finder(manual, 424316, 425567))
self.assertEqual((425298, 432213), finder(manual, 425568, 432213))
