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})