class OLEWriterTest(unittest.TestCase): def setUp(self): self.f = "test.ole" self.o = OLEWriter(self.f) def tearDown(self): self.o.close() self.o = None # todo!!! if os.path.exists(self.f): os.remove(self.f) def test_set_size_too_big(self): err = "Should have raised MaxSizeError" self.assertRaises(Exception, lambda:self.o.set_size(999999999)) def test_booksize_large(self): self.o.set_size(8192) self.assertEqual(8192, self.o._booksize) def test_booksize_small(self): self.o.set_size(2048) self.assertEqual(4096, self.o._booksize) def test_biffsize(self): self.o.set_size(2048) self.assertEqual(2048, self.o._biffsize) def test_size_allowed(self): self.o.set_size(4096) self.assertEqual(True, self.o._size_allowed) def test_big_block_size_default(self): self.o.set_size(4096) self.o._calculate_sizes() self.assertEqual(8, self.o._big_blocks, "Bad big block size") def test_big_block_size_rounded_up(self): self.o.set_size(4099) self.o._calculate_sizes() self.assertEqual(9, self.o._big_blocks, "Bad big block size") def test_list_block_size(self): self.o.set_size(4096) self.o._calculate_sizes() self.assertEqual(1, self.o._list_blocks, "Bad list block size") def test_root_start_size_default(self): self.o.set_size(4096) self.o._calculate_sizes() self.assertEqual(8, self.o._big_blocks, "Bad root start size") def test_root_start_size_rounded_up(self): self.o.set_size(4099) self.o._calculate_sizes() self.assertEqual(9, self.o._big_blocks, "Bad root start size")