def setUp(self): dbfile = ':memory:' self.db = DataBase(dbfile) self.db.create() self.db.insert(2,CURRENT,23.) self.db.insert(3,CURRENT,23.1) self.db.insert(4,CURRENT,23.2)
class TestDB(unittest.TestCase): def setUp(self): dbfile = ':memory:' self.db = DataBase(dbfile) self.db.create() self.db.insert(2,CURRENT,23.) self.db.insert(3,CURRENT,23.1) self.db.insert(4,CURRENT,23.2) def test_iter(self): self.assertEqual(len(list(self.db)),3) def test_range(self): self.assertEqual(len(self.db.getrange(CURRENT,1.5,3.5)),2)