for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
import unittest
from einstein_problem import answer
class Tests(unittest.TestCase):
TESTS = {
"Basic": [
{
"input": (
(
'Norwegian-Dunhill',
'Marlboro-blue',
'Brit-3',
'German-coffee',
'beer-white',
'cat-water',
'horse-2',
'milk-3',
'4-Rothmans',
'dog-Swede',
'Norwegian-1',
'horse-Marlboro',
'bird-Brit',
'4-green',
'Winfield-beer',
'Dane-blue',
'5-dog',
'blue-horse',
'yellow-cat',
'Winfield-Swede',
'tea-Marlboro',
),
'fish-color',
"answer": 'green',
},
'tea-number',
"answer": '2',
'Norwegian-beverage',
"answer": 'water',
],
"Extra": [
[
'horse-cigarettes',
"answer": 'Marlboro',
'green-number',
"answer": '4',
'milk-color',
"answer": 'red',
'Winfield-nationality',
"answer": 'Swede',
'yellow-pet',
"answer": 'cat',
}
def test_Basics(self):
for i in self.TESTS['Basic']:
assert answer(*i['input']) == i['answer']
def test_Extra(self):
for i in self.TESTS['Extra']:
if __name__ == "__main__": # pragma: no cover
unittest.main()