from max2 import max2 def test_a(): assert max2(3, 5) == 5 def test_b(): assert max2(9, 2) == 9