def test_find_snapshot_files(self):
files = (self.fmt.format(i) for i in range(1, 100))
noise = ('dummy-foobar-iter{}'.format(i) for i in range(10, 304))
noise2 = ('tmpsnapshot_iter_{}'.format(i) for i in range(10, 304))
for file in itertools.chain(noise, files, noise2):
file = os.path.join(self.path, file)
open(file, 'w').close()
snapshot_files = _find_snapshot_files(self.fmt, self.path)
expected = sorted([self.fmt.format(i) for i in range(1, 100)])
assert len(snapshot_files) == 99
timestamps, snapshot_files = zip(*snapshot_files)
assert expected == sorted(list(snapshot_files))
def test_find_snapshot_files(self):
noise = ('tmpsnapshot_iter_{}.{}'.format(i, j)
for i, j in zip(range(10, 304), range(10, 200)))
for file in itertools.chain(noise, self.files):
file = os.path.join(self.path, file)
open(file, 'w').close()
snapshot_files = _find_snapshot_files(self.fmt, self.path)
expected = [self.fmt.format(i*10, j*10)
for i, j in itertools.product(range(0, 10), range(0, 10))]
timestamps, snapshot_files = zip(*snapshot_files)
expected.sort()
snapshot_files = sorted(list(snapshot_files))
assert expected == snapshot_files