fast_comparator_test.py

import dis
import unittest

from humu.storage.impl.fast_comparator import Comparison, makeFastComparator


class testFastComparator(unittest.TestCase):
    def testSimpleComparator(self):
        # This should make a function
        # def compareTwo(x1, x2):
        #   return x2 == b'1234' and x1 > b'0000'
        compareTwo = makeFastComparator(
            2,
            [
                Comparison(1, dis.cmp_op.index('=='), b'1234'),
                Comparison(0, dis.cmp_op.index('>'), b'0000'),
            ],
        )
        self.assertTrue(compareTwo(b'5555', b'1234'))
        # Fail the first test
        self.assertFalse(compareTwo(b'5555', b'1000'))
        # Fail the second test
        self.assertFalse(compareTwo(b'****', b'1234'))

    def testCheckNothing(self):
        noTestsCompare = makeFastComparator(2, [])
        self.assertTrue(noTestsCompare('asdfa', 'dfhgdf'))

    def testCompareThree(self):
        compareThree = makeFastComparator(
            3,
            [
                Comparison(0, dis.cmp_op.index('=='), b'1234'),
                Comparison(1, dis.cmp_op.index('=='), b'2345'),
                Comparison(2, dis.cmp_op.index('=='), b'3456'),
            ],
        )
        self.assertTrue(compareThree(b'1234', b'2345', b'3456'))
        self.assertFalse(compareThree(b'0000', b'2345', b'3456'))
        self.assertFalse(compareThree(b'1234', b'0000', b'3456'))
        self.assertFalse(compareThree(b'1234', b'2345', b'0000'))

    def testCompareList(self):
        compareThree = makeFastComparator(
            3,
            [
                Comparison(0, dis.cmp_op.index('in'), [b'1234', b'2345', b'3456']),
                Comparison(1, dis.cmp_op.index('in'), [b'1234']),
                Comparison(2, dis.cmp_op.index('in'), [b'2345', b'3456']),
            ],
        )
        self.assertTrue(compareThree(b'1234', b'1234', b'2345'))
        self.assertTrue(compareThree(b'2345', b'1234', b'3456'))
        self.assertFalse(compareThree(b'0000', b'1234', b'3456'))
        self.assertFalse(compareThree(b'1234', b'1234', b'1234'))
        self.assertFalse(compareThree(b'1234', b'1234', b'0000'))


if __name__ == '__main__':
    unittest.main()