def test_response_for_user(self):
"Returns correct data for one user."
photos = ditto_flickr.annual_photo_counts(nsid='1234567890@N01')
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 3)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_count_by_response(self):
"Returns correct data for all users with count_by='taken_time'."
photos = ditto_flickr.annual_photo_counts(count_by='taken_time')
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 4)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_default_response(self):
"Returns correct data for all users."
photos = ditto_flickr.annual_photo_counts()
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 4)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_count_by_response(self):
"Returns correct data for all users with count_by='taken_time'."
photos = ditto_flickr.annual_photo_counts(count_by='taken_time')
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 4)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_response_for_user(self):
"Returns correct data for one user."
photos = ditto_flickr.annual_photo_counts(nsid='1234567890@N01')
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 3)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_default_response(self):
"Returns correct data for all users."
photos = ditto_flickr.annual_photo_counts()
self.assertEqual(len(photos), 2)
self.assertEqual(photos[0]['year'], 2015)
self.assertEqual(photos[0]['count'], 4)
self.assertEqual(photos[1]['year'], 2016)
self.assertEqual(photos[1]['count'], 2)
def test_empty_years(self):
"It should include years for which there are no photos."
# Add a photo in 2018, leaving a gap for 2017:
PhotoFactory(taken_time=datetime_from_str('2018-01-01 12:00:00'),
user=self.user_1)
photos = ditto_flickr.annual_photo_counts(count_by='taken_time')
self.assertEqual(len(photos), 4)
self.assertEqual(photos[2]['year'], 2017)
self.assertEqual(photos[2]['count'], 0)
def test_empty_years(self):
"It should include years for which there are no photos."
# Add a photo in 2018, leaving a gap for 2017:
PhotoFactory(taken_time=datetime_from_str('2018-01-01 12:00:00'),
user=self.user_1)
photos = ditto_flickr.annual_photo_counts(count_by='taken_time')
self.assertEqual(len(photos), 4)
self.assertEqual(photos[2]['year'], 2017)
self.assertEqual(photos[2]['count'], 0)