def test_get_user_info(self): email = '*****@*****.**' log.info(f'Downloading user info for account {email}...') fpl = FPLPandas(email, 'TestMcTestFace') user_info = fpl.get_user_info() log.info(f'User info downloaded.') self.assertTrue(user_info.shape[0] > 0)
def test_get_user_info(self): test_data = {'player': {'entry': '123'}} expected_df = pd.DataFrame.from_dict([test_data['player']]) fpl_mock = mock.MagicMock() async def mock_login(email, password): self.assertEqual(email, 'email') self.assertEqual(password, 'password') async def mock_get_user_info(): return {'player': {'entry': '123'}} fpl_mock.get_user_info = mock_get_user_info fpl_mock.login = mock_login fpl = FPLPandas('email', 'password', fpl=fpl_mock) actual_df = fpl.get_user_info() assert_frame_equal(expected_df, actual_df)