package addtion help

[PyMan]

like this following example how can i do addition program.

1 2 3 4 5 6 7 8 9 10 11

class Mammals:     def __init__(self):         ''' Constructor for this class. '''         # Create some member animals         self.members = ['Tiger', 'Elephant', 'Wild Cat']           def printMembers(self):         print('Printing members of the Mammals class')         for member in self.members:             print('\t%s ' % member)

1 2 3 4 5 6

from Animals import Mammals from Animals import Birds      myMammal = Mammals() myMammal.printMembers()

like this i want to create addition,sub program where i type following program for addition but i want to put 2 addition numbers in test.py programs insted of taking 2 no's in shell.

1 2 3 4 5 6 7 8

class addition:     def add():         a=int(input("Enter the number : "))         b=int(input("Enter the number two : "))         c= a + b         print(c)   obj = addition.add()

what i need to type in test file ?

[Gribouillis]

I wrote a way to do this by modifying the input() builtin function: suppose you have the module

1 2 3 4 5 6 7 8

# theaddition.py   class addition:     def add():         a=int(input("Enter the number : "))         b=int(input("Enter the number two : "))         c= a + b         print(c)

Then you write the following test file

1 2 3 4 5 6 7

# testaddition.py   from fakeinput import fakeinput from theaddition import addition   with fakeinput(['12', '19']):     obj = addition.add()

Then you get the following output

Output:
λ python3 testaddition.py
Enter the number : 12
Enter the number two : 19
31

To make this work, you need to save the following file

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32

# fakeinput.py   import contextlib import sys   @contextlib.contextmanager def fakeinput(sequence):     inp = MyInput(sequence)     try:         yield     finally:         inp.close()   class MyInput:     def __init__(self, seq):         self.seq = ('{}'.format(x) for x in seq)         self.old = sys.modules['builtins'].input         sys.modules['builtins'].input = self         self.closed = False     def __call__(self, s):         try:             v = next(self.seq)         except StopIteration:             self.close()             return input(s)         else:             print(s, v, sep='')             return v     def close(self):         if not self.closed:             sys.modules['builtins'].input = self.old             self.closed = True