@@ -3,7 +3,7 @@
 📝 Instructions:
 
 **Statement**
-Write a program that takes three numbers and prints their sum. Every number is given on a separate line.
+Taking 3 numbers from the input, print their sum. Every number is given on a separate line.
 
 **Example input**
 - 2

@@ -2,5 +2,4 @@
 first_number = input("First input")
 second_number = input("Second input")
 third_number = input("Third input")
-
-# Invoke the function with any three numbers
+# print here the sum of three inputs

@@ -1,19 +1,22 @@
-import pytest, io, sys, json
-
+import pytest, io, sys, json, mock, re, os
+path = os.path.dirname(os.path.abspath(__file__))+'/app.py'
 
+@pytest.mark.it("You need to use the print() method")
+def test_use_print():
+with open(path, 'r') as content_file:
+content = content_file.read()
+regex = re.compile(r"print(\s)*\(")
+assert bool(regex.search(content)) == True
 
 @pytest.mark.it('Sum all three input numbers and print on the console the result')
-def test_add_variables(mocker, stdin):
-
-_stdin = json.loads(stdin)
-mocker.('builtins.input', lambda x: _stdin.pop(0))
+def test_add_variables(capsys, app):
 
-sys.stdout = buffer = io.StringIO()
-import app
+fake_input = [2,3,4] #fake input
+with mock.('builtins.input', lambda x: fake_input.pop()):
+app()
+captured = capsys.readouterr()
+assert captured.out == "9\n"
 
-_stdin = json.loads(stdin)
-sumatory = int(_stdin[0]) + int(_stdin[1]) + int(_stdin[2])
 
-assert buffer.getvalue() == str(sumatory)+"\n"