36-net-amount

Author: kiddopro

exercises/36-net-amount/solution.hide.py

@@ -1,6 +1,6 @@
 netAmount = 0
 while True:
-    s = raw_input()
+    s = input("")
     if not s:
         break
     values = s.split(" ")
@@ -12,4 +12,4 @@
         netAmount-=amount
     else:
         pass
-print netAmount
+print (netAmount)

exercises/36-net-amount/test.py

@@ -1,3 +1,11 @@
-import pytest, io, sys, json, mock, re, os, app
+import pytest, io, sys, json, mock, re, os
 path = os.path.dirname(os.path.abspath(__file__))+'/app.py'
 
+@pytest.mark.it('The solution should return the expected output')
+def test_output(capsys, app):
+    fake_input = ["D 300 D 300 W 200 D 100"] #fake input
+    with mock.patch('builtins.input', lambda x: fake_input.pop()):
+        app()
+        captured = capsys.readouterr()
+        assert captured.out == "500\n"
+