Assert Called With Pytest Framework