def testReadBadCount(self):
handle = StringIO("""BEAT_START
COUNT |^e+d|
BEAT_END""")
iterator = fileUtils.dbFileIterator(handle)
self.assertRaises(DBErrors.BadCount,
dbfsv0.BeatStructureV0().read, iterator)
def testWriteFullBeat(self):
beat = Beat.Beat(Counter.Counter("e+a"))
handle = StringIO()
indenter = fileUtils.Indenter(handle)
dbfsv0.BeatStructureV0().write(beat, indenter)
output = handle.getvalue().splitlines()
self.assertEqual(output, ["BEAT_START", " COUNT |^e+a|", "BEAT_END"])
def testReadFull(self):
handle = StringIO("""BEAT_START
COUNT |^e+a|
BEAT_END""")
iterator = fileUtils.dbFileIterator(handle)
beat = dbfsv0.BeatStructureV0().read(iterator)
self.assertEqual("".join(beat.count(1)), "1e+a")
def testReadBadTicks(self):
handle = StringIO("""BEAT_START
NUM_TICKS x
COUNT |^e+a|
BEAT_END""")
iterator = fileUtils.dbFileIterator(handle)
self.assertRaises(DBErrors.InvalidInteger,
dbfsv0.BeatStructureV0().read, iterator)
def testReadBadLine(self):
handle = StringIO("""BEAT_START
COUNT |^e+a|
BAD_LINE xxx
BEAT_END""")
iterator = fileUtils.dbFileIterator(handle)
self.assertRaises(DBErrors.UnrecognisedLine,
dbfsv0.BeatStructureV0().read, iterator)
