def testDequePickle(self):
benchmark_deque = Deque(5)
with tempfile.NamedTemporaryFile('w+b', delete=False) as f:
pickle.dump(benchmark_deque, f)
with open(f.name, 'rb') as f2:
pickled_deque = pickle.load(f2)
self.assertEqual(benchmark_deque, pickled_deque)
os.unlink(f.name)
def testDequeIdx(self):
deque1 = deque(maxlen=10)
deque2 = Deque(10)
values = np.random.randn(1000)
for v in values:
deque1.append(v)
deque2.dumps([v])
max_val = max(deque1)
expected_idx = np.where(np.array(deque1) == max_val)[0][0]
caclulated_idx = deque2.idx(max_val)
self.assertEqual(expected_idx, caclulated_idx)
def testDequeDeepCopy(self):
benchmark_deque = Deque(5)
copied_deque = copy.deepcopy(benchmark_deque)
self.assertEqual(copied_deque, benchmark_deque)
def testDequeSum(self):
deque = Deque(5)
deque.dumps([1., 2., 3., 4., 5., 6.])
self.assertEqual(deque.sum(), 20.)