def test_bed_number_method(self): """ Test that the bed number generator works """ gen = UsersGenerator({}) bed_number_gen = gen.get_beds_number_generator(2) bed_number_first = bed_number_gen.next() bed_number_last = bed_number_gen.next() self.assertEqual(bed_number_first, 1, 'Incorrect first generator call') self.assertEqual(bed_number_last, 2, 'Incorrect last generator call')