def setUp(self):
        super(TestDoctorAllocationDeallocate, self).setUp()

        helpers.create_test_data(self, self.cr, self.uid, iterations=2)
        self.ward_manager_id = self.users['wm'][0]
        self.ward_id = self.locations.keys()[0]
        self.ward_bed_ids = self.locations[self.ward_id]
        self.ward_all_location_ids = [self.ward_id] + self.ward_bed_ids

        self.doctor_allocation_wizard_id = self.doctor_allocation.create(
            self.cr, self.ward_manager_id,
            {'location_ids': [(6, 0, self.ward_all_location_ids)]})
    def setUp(self):
        super(TestDoctorAllocationDeallocate, self).setUp()

        helpers.create_test_data(self, self.cr, self.uid, iterations=2)
        self.ward_manager_id = self.users['wm'][0]
        self.ward_id = self.locations.keys()[0]
        self.ward_bed_ids = self.locations[self.ward_id]
        self.ward_all_location_ids = [self.ward_id] + self.ward_bed_ids

        self.doctor_allocation_wizard_id = self.doctor_allocation.create(
            self.cr, self.ward_manager_id,
            {'location_ids': [(6, 0, self.ward_all_location_ids)]}
        )
 def setUpClass(cls):
     super(TestAllocationWizards, cls).setUpClass()
     cr, uid, = cls.cr, cls.uid
     helpers.create_test_data(cls, cr, uid)