def test_get_lut_card_in_what_range_idxs(self):
lg = _LutGetterLeduc(env_cls=StandardLeduc)
lut = lg.get_card_in_what_range_idxs_LUT()
assert lut.shape == (6, 1)
assert not np.any(lut == -2)
counts = np.zeros(6, np.int32)
for c in range(6):
for h in lut[c]:
counts[h] += 1
assert np.all(counts == 1)
def test_get_hole_card_2_idx_lut(self):
lg = _LutGetterLeduc(env_cls=StandardLeduc)
lut = lg.get_hole_card_2_idx_LUT()
assert lut.shape == (6, 1)
assert not np.any(lut == -2)
def test_get_2d_card_2_1d_card_lut(self):
lg = _LutGetterLeduc(env_cls=StandardLeduc)
lut = lg.get_2d_card_2_1d_card_LUT()
assert lut.shape == (3, 2)
assert not np.any(lut == -2)