def test_getRating(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test")
imageDatabase.addRating("test", "5")
imageDatabase.addRating("test", "3")
rating = imageDatabase.getRating("test")
self.assertEqual(rating, 4)
def test_getRatings(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test")
imageDatabase.addRating("test", "4")
imageDatabase.addRating("test", "3")
ratings = imageDatabase.getRatings("test")
numRatings = len(ratings)
self.assertEqual(numRatings, 2)
def test_getImages(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test1")
imageDatabase.addImage("test2")
imageDatabase.addImage("test3")
numImages = imageDatabase.getImages()
self.assertEqual(len(numImages), 3)
def test_wipeDatabase(self):
imageDatabase.addImage("test1")
imageDatabase.addImage("test2")
imageDatabase.addImage("test3")
imageDatabase.wipeDatabase()
images = imageDatabase.getImages()
numImages = len(images)
self.assertEqual(numImages, 0)
def test_returnImage(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test1")
imageDatabase.addImage("test2")
self.assertEqual(imageDatabase.returnImage("test1")['image'], "test1")
def test_removeImage(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test")
imageDatabase.removeImage("test")
self.assertEqual(imageDatabase.returnImage("test"), None)
def test_addRating(self):
imageDatabase.wipeDatabase()
imageDatabase.addImage("test")
imageDatabase.addRating("test", "4")
ratings = imageDatabase.returnImage("test")['ratings']
self.assertEqual(ratings[0], "4")
def test_addImage(self):
imageDatabase.addImage("test")
self.assertEqual(imageDatabase.returnImage("test")['image'], "test")
