def test_segments(self): f = self.sample_daf() d = DAF(f) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 2) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128)
def test_segments(self): f = self.sample_daf() d = DAF(f) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 2) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128)
def test_add_segment_when_summary_block_is_full(self): f = self.sample_daf() d = DAF(f) # Update n_summaries of final summary block to full. d.file.seek(6 * 1024 + 16) d.file.write(Struct('d').pack(d.summaries_per_record)) d.add_array(b'Summary Name 3', (121.0, 232.0, 343), [3003.0] * 200) # Reset n_summaries of that block back to its real value. d.file.seek(6 * 1024 + 16) d.file.write(Struct('d').pack(1)) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 3) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq(summaries[2], (b'Summary Name 3', (121.0, 232.0, 343, 1281, 1480))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128) eq(list(d.map(summaries[2][1])), [3003.0] * 200)
def test_header(self): f = self.sample_daf() d = DAF(f) eq = self.assertEqual eq(d.locidw, b'DAF/SPK') eq(d.nd, 2) eq(d.ni, 3) eq(d.locifn_text, b'Internal Name') eq(d.fward, 3) eq(d.bward, 7) eq(d.free, 0x401) eq(d.locfmt, b'LTL-IEEE')
def test_add_segment_when_summary_block_is_full(self): f = self.sample_daf() d = DAF(f) # Update n_summaries of final summary block to full. d.file.seek(6 * 1024 + 16) d.file.write(Struct('d').pack(d.summaries_per_record)) d.add_array(b'Summary Name 3', (121.0, 232.0, 343), [3003.0] * 200) # Reset n_summaries of that block back to its real value. d.file.seek(6 * 1024 + 16) d.file.write(Struct('d').pack(1)) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 3) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq(summaries[2], (b'Summary Name 3', (121.0, 232.0, 343, 1281, 1480))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128) eq(list(d.map(summaries[2][1])), [3003.0] * 200)
def test_add_segment(self): f = self.sample_daf() d = DAF(f) d.add_array(b'Summary Name 3', (121.0, 232.0, 343), [3003.0] * 128) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 3) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq(summaries[2], (b'Summary Name 3', (121.0, 232.0, 343, 1025, 1152))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128) eq(list(d.map(summaries[2][1])), [3003.0] * 128)
def get_summary(url, spk=True): ''' simple function to retrieve the header of a BSP file and return SPK object''' # connect to file at URL bspurl = urllib2.urlopen(url) # retrieve the "tip" of a file at URL bsptip = bspurl.read(10**5) # first 100kB # save data in fake file object (in-memory) bspstr = StringIO(bsptip) # load into DAF object daf = DAF(bspstr) # return either SPK or DAF object if spk: # make a SPK object spk = SPK(daf) # return representation return spk else: # return representation return daf
def test_add_segment(self): f = self.sample_daf() d = DAF(f) d.add_array(b'Summary Name 3', (121.0, 232.0, 343), [3003.0] * 128) summaries = list(d.summaries()) eq = self.assertEqual eq(len(summaries), 3) eq(summaries[0], (b'Summary Name 1', (101.0, 202.0, 303, 513, 640))) eq(summaries[1], (b'Summary Name 2', (111.0, 222.0, 333, 641, 768))) eq(summaries[2], (b'Summary Name 3', (121.0, 232.0, 343, 1025, 1152))) eq = self.assertSequenceEqual eq(list(d.map(summaries[0][1])), [1001.0] * 128) eq(list(d.map(summaries[1][1])), [2002.0] * 128) eq(list(d.map(summaries[2][1])), [3003.0] * 128)
def open(cls, path): """Open the file at `path` and return an SPK instance. """ return cls(DAF(open(path, 'rb')))