def test_parse_image_size(self): # Confirm things work as expected with a sane value result = generate_iso._parse_image_size(100) self.assertEqual(100 * 1024 * 1024, result)
def test_none(self): # Assert that if None is received, the size is set to a dvd-sized ISO result = generate_iso._parse_image_size(None) self.assertEqual(generate_iso.DVD_ISO_SIZE * 1024 * 1024, result)