墨子 Code

@@ -41,9 +41,9 @@

41 def hold_lot(lot, reason):

42 log.info("hold %s", lot.id)

43− for wafer_id in lot.wafer_ids:

44− wafer = db.get_wafer(wafer_id)

45− wafer.hold(reason)

43+ wafers = db.get_wafers(lot.wafer_ids)

44+ db.hold_wafers(wafers, reason,

45+ batch_size=200)

46 audit.record(lot, "HOLD", reason)

47 return HoldResult(lot.id, len(wafers))

tests/test_lot_hold_large.py

1+def test_hold_800_wafers(db, make_lot):

2+ lot = make_lot(wafers=800)

3+ result = hold_lot(lot, "PM overdue")

4+ assert result.count == 800

5+ assert db.queries < 10