Esempio n. 1
0
    def setUp(self):
        self.dirPath = "files/testing"
        if not os.path.isdir(self.dirPath):
            os.makedirs(self.dirPath)
            self._createdDirs = True
        else:
            self._createdDirs = False

        self.blarg_index = TreeIndex(database_name='testing',
                                     fieldName='blarg')
        self.blarg_index.insert(1, 'ts-1')
        self.blarg_index.insert(1, 'ts-2')
        self.blarg_index.insert(1, 'ts-3')
        self.blarg_index.insert(1, 'ts-4')
        self.blarg_index.insert(1, 'ts-5')
        self.blarg_index.insert(2, 'ts-6')
        self.blarg_index.insert(7, 'ts-7')
        self.blarg_index.insert(8, 'ts-8')
        self.blarg_index.insert(2, 'ts-9')
        self.blarg_index.insert(8, 'ts-10')
    def setUp(self):
        self.dirPath = "files/testing"
        if not os.path.isdir(self.dirPath):
            os.makedirs(self.dirPath)
            self._createdDirs = True
        else:
            self._createdDirs = False

        self.blarg_index = TreeIndex(database_name="testing", fieldName="blarg")
        self.blarg_index.insert(1, "ts-1")
        self.blarg_index.insert(1, "ts-2")
        self.blarg_index.insert(1, "ts-3")
        self.blarg_index.insert(1, "ts-4")
        self.blarg_index.insert(1, "ts-5")
        self.blarg_index.insert(2, "ts-6")
        self.blarg_index.insert(7, "ts-7")
        self.blarg_index.insert(8, "ts-8")
        self.blarg_index.insert(2, "ts-9")
        self.blarg_index.insert(8, "ts-10")
class TreeIndexTests(unittest.TestCase):
    def setUp(self):
        self.dirPath = "files/testing"
        if not os.path.isdir(self.dirPath):
            os.makedirs(self.dirPath)
            self._createdDirs = True
        else:
            self._createdDirs = False

        self.blarg_index = TreeIndex(database_name="testing", fieldName="blarg")
        self.blarg_index.insert(1, "ts-1")
        self.blarg_index.insert(1, "ts-2")
        self.blarg_index.insert(1, "ts-3")
        self.blarg_index.insert(1, "ts-4")
        self.blarg_index.insert(1, "ts-5")
        self.blarg_index.insert(2, "ts-6")
        self.blarg_index.insert(7, "ts-7")
        self.blarg_index.insert(8, "ts-8")
        self.blarg_index.insert(2, "ts-9")
        self.blarg_index.insert(8, "ts-10")

    def tearDown(self):
        self.blarg_index.deleteIndex()
        if self._createdDirs:
            os.removedirs(self.dirPath)

    def test_get(self):
        self.assertEqual(set(self.blarg_index.getEqual(8)), set(["ts-8", "ts-10"]))

    def test_getHigher(self):
        self.assertEqual(set(self.blarg_index.getHigherThan(6)), set(["ts-7", "ts-8", "ts-10"]))

    def test_getLower(self):
        self.assertEqual(set(self.blarg_index.getLowerThan(2)), set(["ts-1", "ts-2", "ts-3", "ts-4", "ts-5"]))

    def test_missingField(self):
        self.assertEqual(set(self.blarg_index.getEqual(11)), set([]))

    def test_remove(self):
        self.assertEqual(set(self.blarg_index.getEqual(8)), set(["ts-8", "ts-10"]))
        self.blarg_index.remove(fieldValue=8, pk="ts-8")
        self.assertEqual(set(self.blarg_index.getEqual(8)), set(["ts-10"]))

    def test_missingPK(self):
        with self.assertRaises(ValueError):
            self.blarg_index.remove(fieldValue=8, pk="ts-100")
Esempio n. 4
0
class TreeIndexTests(unittest.TestCase):
    def setUp(self):
        self.dirPath = "files/testing"
        if not os.path.isdir(self.dirPath):
            os.makedirs(self.dirPath)
            self._createdDirs = True
        else:
            self._createdDirs = False

        self.blarg_index = TreeIndex(database_name='testing',
                                     fieldName='blarg')
        self.blarg_index.insert(1, 'ts-1')
        self.blarg_index.insert(1, 'ts-2')
        self.blarg_index.insert(1, 'ts-3')
        self.blarg_index.insert(1, 'ts-4')
        self.blarg_index.insert(1, 'ts-5')
        self.blarg_index.insert(2, 'ts-6')
        self.blarg_index.insert(7, 'ts-7')
        self.blarg_index.insert(8, 'ts-8')
        self.blarg_index.insert(2, 'ts-9')
        self.blarg_index.insert(8, 'ts-10')

    def tearDown(self):
        self.blarg_index.deleteIndex()
        if self._createdDirs:
            os.removedirs(self.dirPath)

    def test_get(self):
        self.assertEqual(set(self.blarg_index.getEqual(8)),
                         set(['ts-8', 'ts-10']))

    def test_getHigher(self):
        self.assertEqual(set(self.blarg_index.getHigherThan(6)),
                         set(['ts-7', 'ts-8', 'ts-10']))

    def test_getLower(self):
        self.assertEqual(set(self.blarg_index.getLowerThan(2)),
                         set(['ts-1', 'ts-2', 'ts-3', 'ts-4', 'ts-5']))

    def test_missingField(self):
        self.assertEqual(set(self.blarg_index.getEqual(11)), set([]))

    def test_remove(self):
        self.assertEqual(set(self.blarg_index.getEqual(8)),
                         set(['ts-8', 'ts-10']))
        self.blarg_index.remove(fieldValue=8, pk='ts-8')
        self.assertEqual(set(self.blarg_index.getEqual(8)), set(['ts-10']))

    def test_missingPK(self):
        with self.assertRaises(ValueError):
            self.blarg_index.remove(fieldValue=8, pk='ts-100')