def test_cmp_nbr_headers(self): """ Test to check that the number of headers is correct. """ with open(THREE_CHAN_FCNT, 'rb') as fi: (nbr_channel_set_headers, nbr_extended_headers, nbr_external_headers) = rc._cmp_nbr_headers(fi) self.assertEqual(nbr_channel_set_headers, 3) self.assertEqual(nbr_extended_headers, 3) self.assertEqual(nbr_external_headers, 1)