eris/tests/paged_list_test.py

#!/usr/bin/env python3.9


import pickle
import tempfile
import unittest

import eris.paged_list as paged_list


class PagedListTestCase(unittest.TestCase):

    def test_batch(self):
        self.assertEqual(list(paged_list.batch(iter([3,4,5,6,7]), 2)),
                         [[3, 4], [5, 6], [7]])

    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])
            self.assertEqual(list(list_), [3, 4, 5, 6, 7, 8])
            self.assertRaises(IndexError, list_.__getitem__, 6)
        with tempfile.TemporaryDirectory() as temp_dir:
            list_ = paged_list.PagedList([], temp_dir, 2, 2)
            self.assertRaises(IndexError, list_.__getitem__, 0)
            # self.assertEqual(list_[3:4], [])   FIX

    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()