def test_friends_ids(samp_auth): """ Test friends/ids method. """ for user_id, _ in TEST_USERS: params = {"user_id": user_id, "count": 10} data, meta = rest.friends_ids(samp_auth, **params) assert meta["code"] == 200 assert len(data["ids"]) >= 10
def test_friends_ids_iter(samp_auth): """ Test friends/ids method w/ cursor_iter. """ for user_id, _ in TEST_USERS: params = {"user_id": user_id, "count": 10, "maxitems": 20} results = [] for data, meta in rest.friends_ids(samp_auth, **params): assert meta["code"] == 200 assert len(data["ids"]) >= 10 results.extend(data["ids"]) assert len(results) >= 20 assert len(set(results)) >= 20
def main(): _, user = sys.argv try: params = {"id": int(user)} except ValueError: params = {"screen_name": user} kfname = "test_keys-twitter.json" auth = read_keys(kfname) data, meta = friends_ids(auth, **params) assert meta["code"] == 200 print(json.dumps(data))