def test_works_well(self): """Make sure that _sched_piece() works well""" idx = (i for i in xrange(64)) sched = _sched_piece(idx, 19, 16, 8) self.assertEquals(next(sched), (0, 0, 8)) self.assertEquals(next(sched), (0, 8, 8)) self.assertEquals(next(sched), (1, 0, 3))
def test_checks_parameters(self): """Make sure it raises RuntimeError if passed invalid parameters""" idx = (i for i in xrange(64)) sched = _sched_piece(idx, 19, 16, 81) self.assertRaises(RuntimeError, next, sched)