'''
Created on 24.02.2017

@author: steinorb
'''
import unittest.mock
import waehlen
import builtins


class WaehlenTest(unittest.TestCase):


    @unittest.mock.patch('builtins.input')
    def testAuswahl(self, mock_input):
        mock_input.return_value = "1"
        auswahl = waehlen.auswahl()
        self.assertEqual(auswahl, 1)


if __name__ == "__main__":
    unittest.main()
@@ -62,21 +62,16 @@
 unittest
-.mock
 %0D%0Aimport
@@ -84,25 +84,8 @@
 en%0D%0A
-import builtins%0D%0A
 %0D%0A%0D%0A
@@ -169,16 +169,50 @@
 .input')
+ # mock the builtin input() method
 %0D%0A    de

simplify import