def test_weighted_dice_set_add_4():
wds1 = WeightedDiceSet(4, 3, weights=[1, 3, 2])
ds2 = DiceSet(3, 3)
wds3 = wds1 + ds2
assert wds3.weights == [2, 4, 3]
def test_weighted_dice_set_add_3():
wds1 = WeightedDiceSet(4, 3, weights=[1, 3, 2])
ds2 = DiceSet(3, 3)
wds3 = wds1 + ds2
assert wds3.number == 7
def test_weighted_dice_set_add_2():
wds1 = WeightedDiceSet(4, 3, weights=[1, 3, 2])
wds2 = WeightedDiceSet(2, 3, weights=[3, 3, 1])
wds3 = wds1 + wds2
assert wds3.weights == [4, 6, 3]
def test_weighted_dice_set_add_1():
wds1 = WeightedDiceSet(4, 3, weights=[1, 3, 2])
wds2 = WeightedDiceSet(2, 3, weights=[3, 3, 1])
wds3 = wds1 + wds2
assert wds3.number == 6
def test_weighted_dice_set_init_2():
wds = WeightedDiceSet(4, 3, weights=[1, 3, 2])
assert wds.weights == [1, 3, 2]
def test_weighted_dice_set_init_1():
wds = WeightedDiceSet(4, 3)
assert wds.weights == [1, 1, 1]