def test_byLocations(self):
self.fileFactory.fromProvenance.side_effect = lambda p: 'img_' + p['a']
from niprov.jsonfile import JsonFile
repo = JsonFile(self.dependencies)
img1 = self.imageWithProvenance({'location': 'i'})
img2 = self.imageWithProvenance({'location': 'j'})
img3 = self.imageWithProvenance({'location': 'm'})
img4 = self.imageWithProvenance({'location': 'f'})
img5 = self.imageWithProvenance({'location': 'x'})
repo.all = Mock()
repo.all.return_value = [img1, img2, img3, img4, img5]
out = repo.byLocations(['j', 'f', 'k'])
self.assertEqual([img2, img4], out)
def test_byLocations(self):
self.fileFactory.fromProvenance.side_effect = lambda p: 'img_'+p['a']
from niprov.jsonfile import JsonFile
repo = JsonFile(self.dependencies)
img1 = self.imageWithProvenance({'location':'i'})
img2 = self.imageWithProvenance({'location':'j'})
img3 = self.imageWithProvenance({'location':'m'})
img4 = self.imageWithProvenance({'location':'f'})
img5 = self.imageWithProvenance({'location':'x'})
repo.all = Mock()
repo.all.return_value = [img1,img2,img3,img4,img5]
out = repo.byLocations(['j','f','k'])
self.assertEqual([img2, img4], out)
