eris/tests/paged_list_test.py
Andrew Hamilton 1cf2fc445c Revert "Move to python3.8.0"
This reverts commit 04fb17cb3f.
- Am seeing a bug where only one worker continues processing
  on python3.8.
- Ubuntu and freedesktop-sdk are still on 3.7.
- Not using any features of 3.8 yet.
- Also removes color from mypy.
2019-12-19 14:47:00 +10:00

39 lines
1.3 KiB
Python
Executable file

#!/usr/bin/env python3.7
# Copyright (C) 2019 Andrew Hamilton. All rights reserved.
# Licensed under the Artistic License 2.0.
import pickle
import tempfile
import unittest
import eris.paged_list as paged_list
class PagedListTestCase(unittest.TestCase):
def test_getitem(self):
with tempfile.TemporaryDirectory() as temp_dir:
list_ = paged_list.PagedList([3, 4, 5, 6], temp_dir, 4, 2)
self.assertEqual(list_[1], 4)
self.assertEqual(list_[1:3], [4, 5])
self.assertEqual(list_[0:4], [3, 4, 5, 6])
with tempfile.TemporaryDirectory() as temp_dir:
list_ = paged_list.PagedList([3, 4, 5, 6], temp_dir, 2, 2)
self.assertEqual(list_[1:3], [4, 5])
with tempfile.TemporaryDirectory() as temp_dir:
list_ = paged_list.PagedList([3, 4, 5, 6, 7, 8], temp_dir, 2, 2)
self.assertEqual(list_[1:5], [4, 5, 6, 7])
self.assertEqual(list_[:2], [3, 4])
self.assertEqual(list_[2:], [5, 6, 7, 8])
def test_pickling(self):
with tempfile.TemporaryDirectory() as temp_dir:
list_ = paged_list.PagedList([3, 4, 5], temp_dir, 2, 2)
list_b = pickle.loads(pickle.dumps(list_))
self.assertEqual(list_b[1], 4)
if __name__ == "__main__":
unittest.main()