def testGetImagePath(self): image_lists = self.dummyImageLists() self.assertEqual('image_dir/somedir/image_one.jpg', retrain.get_image_path( image_lists, 'label_one', 0, 'image_dir', 'training')) self.assertEqual('image_dir/otherdir/image_four.jpg', retrain.get_image_path(image_lists, 'label_two', 1, 'image_dir', 'testing'))
def testGetImagePath(self): image_lists = self.dummyImageLists() self.assertEqual('image_dir/somedir/image_one.jpg', retrain.get_image_path( image_lists, 'label_one', 0, 'image_dir', 'training')) self.assertEqual('image_dir/otherdir/image_four.jpg', retrain.get_image_path(image_lists, 'label_two', 1, 'image_dir', 'testing'))
def testGetImagePath(self): image_lists = self.dummyImageLists() self.assertEqual( "image_dir/somedir/image_one.jpg", retrain.get_image_path(image_lists, "label_one", 0, "image_dir", "training"), ) self.assertEqual( "image_dir/otherdir/image_four.jpg", retrain.get_image_path(image_lists, "label_two", 1, "image_dir", "testing"), )