Learning to use pytest with a HackerRank challenge.

findingPercentage.py

def main():
	n = int(input())
	student_marks = {}
	for _ in range(n):
		name, *line = input().split()
		scores = list(map(float, line))
		student_marks[name] = scores
	query_name = input()

	result = sum(student_marks[name])/n
	print(result/n)

	print('%.2f' % result)

findingPercentage_test.py

'''
Run this to test the findingPercentage.py file.
run : pytest findingPercentage_test01.py
'''


import findingPercentage

def test_app(capsys):
    input_values = [2, 'Harsh 25 26.5 28', 'Anurag 26 28 30', 'Harsh']
 
    def mock_input():
        return input_values.pop(0)
    findingPercentage.input = mock_input
 
    findingPercentage.main()
 
    out, err = capsys.readouterr()
 
    assert out == '26.50'
    assert err == ''