added code related to pytest parameters

Author: krsm

.idea/General-Python.iml

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+# to be treated as a python package
+
+

.idea/misc.xml

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+# to be treated as a python package
+
+

.idea/modules.xml

@@ -12,3 +12,7 @@ def calc_sum(a, b):
 
 def calc_multiply(a, b):
     return a*b
+
+
+def calc_square(n):
+    return n*n

.idea/workspace.xml

@@ -3,9 +3,12 @@
 
 Simple study related to pytest
 
+Use command line pytest -v to run tests
+
 """
 
 from Unitests.pystest_stdy import math_file
+import pytest
 
 
 # first test
@@ -16,6 +19,19 @@ def test_calc_sum():
 
 # second test
 def test_calc_multiply():
-    total = math_file.calc_sum(4, 6)
+    total = math_file.calc_multiply(4, 6)
     assert total == 24
 
+
+# third test
+@pytest.mark.parametrize("test_input, expected_output",
+                         [
+                             (5, 25),
+                             (9, 81),
+                             (10, 100),
+                         ])
+def test_calc_square(test_input, expected_output):
+    result = math_file.calc_square(test_input)
+    assert result == expected_output
+
+