def test_example_1(self):
# arrange
reactions = [
'10 ORE => 10 A', '1 ORE => 1 B', '7 A, 1 B => 1 C',
'7 A, 1 C => 1 D', '7 A, 1 D => 1 E', '7 A, 1 E => 1 FUEL'
]
nf = NanoFactory(reactions)
# act
result = nf.CalculateOre(1, 'FUEL')
# assert
self.assertEqual(result, 31)
def test_example_2(self):
# arrange
reactions = [
'9 ORE => 2 A', '8 ORE => 3 B', '7 ORE => 5 C', '3 A, 4 B => 1 AB',
'5 B, 7 C => 1 BC', '4 C, 1 A => 1 CA',
'2 AB, 3 BC, 4 CA => 1 FUEL'
]
nf = NanoFactory(reactions)
# act
result = nf.CalculateOre(1, 'FUEL')
# assert
self.assertEqual(result, 165)