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(sched.next(), (0, 0, 8)) self.assertEquals(sched.next(), (0, 8, 8)) self.assertEquals(sched.next(), (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, sched.next)