def test_load_all_activities(self, mongo_mock):
mongo_mock.db.activities.find.return_value = iter([{'id': '123', 'strava_id': 456, 'average_speed': 20.1},
{'id': '124', 'strava_id': 457, 'average_speed': 30.1}])
result = load_activities(mongo_mock)
self.assertIs(type(result), pd.DataFrame)
self.assertEqual(list(result.id.values), ['123', '124'])
self.assertEqual(list(result.strava_id.values), [456, 457])
self.assertEqual(list(result.average_speed.values), [20.1, 30.1])
def test_load_activities_with_complex_filter(self, mongo_mock):
load_activities(mongo_mock, athlete_id=123, ride_type={'$exists': False})
mongo_mock.db.activities.find.assert_called_with({'athlete_id': 123, 'ride_type': {'$exists': False}})
def test_load_activities_with_simple_filter(self, mongo_mock):
load_activities(mongo_mock, strava_id=456)
mongo_mock.db.activities.find.assert_called_with({'strava_id': 456})
