TestIoVSet Class Reference

Inheritance diagram for TestIoVSet:

Public Member Functions
def	test_add (self)
def	test_remove (self)
def	test_intersection (self)

Detailed Description

Helper class to test a set of IoVs

Definition at line 122 of file test_iov.py.

Member Function Documentation

test_add()

def test_add

(

self

)

Test adding iovs to a set

Definition at line 125 of file test_iov.py.

    def test_add(self):
        """Test adding iovs to a set"""
        inputs = [
            [(0, 0, 0, -1), (1, 0, 1, -1), (2, 1, 2, -1), (3, 1, 3, 1), (3, 2, 3, 2), (3, 3, 3, 6), (3, 8, 3, 8)],
            [(0, i, 0, i) for i in range(0, 100)],
            [(0, i, 0, i) for i in range(0, 100, 2)],
        ]
        results = [
            [(0, 0, 1, -1), (2, 1, 2, -1), (3, 1, 3, 6), (3, 8, 3, 8)],
            [(0, 0, 0, 99)],
            [(0, i, 0, i) for i in range(0, 100, 2)],
        ]
 
        for _ in range(10):
            for input, output in zip(inputs, results):
                s = IoVSet()
                for iov in random.sample(input, len(input)):
                    s.add(iov)
                output = {IntervalOfValidity(e) for e in output}
                self.assertEqual(s.iovs, output)
 

test_intersection()

def test_intersection

(

self

)

Test intersecting two sets

Definition at line 158 of file test_iov.py.

    def test_intersection(self):
        """Test intersecting two sets"""
        iovs = []
        for i in range(6):
            iovs.append((i, 0, i + 5, -1))
        result = {IntervalOfValidity(5, 0, 5, -1)}
        for _ in range(10):
            a = IoVSet([IntervalOfValidity.always()])
            for iov in random.sample(iovs, len(iovs)):
                a = a.intersect(iov)
            self.assertEqual(a.iovs, result)
        full = IoVSet(iovs, allow_overlaps=True)
        self.assertEqual(full.iovs, {IntervalOfValidity(0, 0, 10, -1)})
 
 

test_remove()

def test_remove

(

self

)

Test removing iovs from a set

Definition at line 146 of file test_iov.py.

    def test_remove(self):
        """Test removing iovs from a set"""
        s = IoVSet()
        s.add((0, 0, 0, 100))
        to_remove = [(0, i, 0, i) for i in range(0, 100, 2)]
        result = {IntervalOfValidity(0, i, 0, i) for i in range(1, 99, 2)}
        result.add(IntervalOfValidity(0, 99, 0, 100))
        for _ in range(10):
            for iov in random.sample(to_remove, len(to_remove)):
                s.remove(iov)
            self.assertEqual(s.iovs, result)
 

---

The documentation for this class was generated from the following file:

• framework/tests/test_iov.py