def setUp(self): codes = ['http://twitter.com/' + "".join(sample(letters, self.code_length)) for dummy in range(self.sticker_count)] lines = [choice(self.first_names) + ' ' + choice(self.last_names) for dummy in range(self.sticker_count)] stickers = [Sticker(c, l) for c, l in zip(codes, lines)] self.maker = PDFMaker(self.out_file, stickers)
class PDFMakerTest(unittest.TestCase): sticker_count = 100 code_length = 25 out_file = 'pdfmakertest.pdf' first_names = [ 'Simon', 'Åsa', 'Torbjörn', 'Gåsa', 'Lännart', 'Älin', 'Örban', ] last_names = [ 'Anderssån', 'Erikssån', 'Svänzzon', 'Örnvinge', 'Ågren', 'Silversköld', ] def setUp(self): codes = ['http://twitter.com/' + "".join(sample(letters, self.code_length)) for dummy in range(self.sticker_count)] lines = [choice(self.first_names) + ' ' + choice(self.last_names) for dummy in range(self.sticker_count)] stickers = [Sticker(c, l) for c, l in zip(codes, lines)] self.maker = PDFMaker(self.out_file, stickers) def test_build(self): """Make sure that the out file exists and that it is a PDF file.""" self.maker.build() self.assertTrue(os.path.isfile(self.out_file)) with open(self.out_file, 'r') as f: header_line = f.readlines()[0] self.assertTrue(header_line.startswith('%PDF-'))