def test_citation_type_to_generator(self):
citation_type = 'STATUTES_AT_LARGE'
layer = ExternalCitationLayer(None)
generator = layer.citation_type_to_generator(citation_type)
self.assertEqual(generator,
ExternalCitationLayer.generate_statutes_at_large_link)
citation_type = 'PUBLIC_LAW'
generator = layer.citation_type_to_generator(citation_type)
self.assertEqual(generator,
ExternalCitationLayer.generate_public_law_link)
