def test_market_issuance_acct_reverse(market: OneSidedMarket, last_offer_size, traded_energy):
offer1 = market.offer(10, 20, 'A')
offer2 = market.offer(10, 10, 'A')
offer3 = market.offer(10, last_offer_size, 'D')
market.accept_offer(offer1, 'B')
market.accept_offer(offer2, 'C')
market.accept_offer(offer3, 'A')
assert market.traded_energy['A'] == traded_energy
def test_market_issuance_acct_reverse(last_offer_size, traded_energy):
market = OneSidedMarket(time_slot=DateTime.now())
offer1 = market.offer(10, 20, 'A', 'A')
offer2 = market.offer(10, 10, 'A', 'A')
offer3 = market.offer(10, last_offer_size, 'D', 'D')
market.accept_offer(offer1, 'B')
market.accept_offer(offer2, 'C')
market.accept_offer(offer3, 'A')
assert market.traded_energy['A'] == traded_energy
def test_market_issuance_acct_reverse(last_offer_size, traded_energy):
market = OneSidedMarket(bc=NonBlockchainInterface(str(uuid4())), time_slot=now())
offer1 = market.offer(10, 20, "A", "A")
offer2 = market.offer(10, 10, "A", "A")
offer3 = market.offer(10, last_offer_size, "D", "D")
market.accept_offer(offer1, "B")
market.accept_offer(offer2, "C")
market.accept_offer(offer3, "A")
assert market.traded_energy["A"] == traded_energy
def test_market_acct_simple(market: OneSidedMarket):
offer = market.offer(20, 10, 'A')
market.accept_offer(offer, 'B')
assert market.traded_energy['A'] == offer.energy
assert market.traded_energy['B'] == -offer.energy
assert market.bought_energy('A') == 0
assert market.bought_energy('B') == offer.energy
assert market.sold_energy('A') == offer.energy
assert market.sold_energy('B') == 0
class MarketStateMachine(RuleBasedStateMachine):
offers = Bundle('Offers')
actors = Bundle('Actors')
def __init__(self):
self.market = OneSidedMarket(
area=FakeArea(name='my_fake_house', transfer_fee_pct=0))
super().__init__()
@rule(target=actors,
actor=st.text(min_size=1,
max_size=3,
alphabet=string.ascii_letters + string.digits))
def new_actor(self, actor):
return actor
@rule(target=offers,
seller=actors,
energy=st.integers(min_value=1),
price=st.integers())
def offer(self, seller, energy, price):
return self.market.offer(price, energy, seller)
@rule(offer=offers, buyer=actors)
def trade(self, offer, buyer):
assume(offer.id in self.market.offers)
self.market.accept_offer(offer, buyer)
@precondition(lambda self: self.market.offers)
@rule()
def check_avg_offer_price(self):
price = sum(o.price for o in self.market.offers.values())
energy = sum(o.energy for o in self.market.offers.values())
assert self.market.avg_offer_price == round(price / energy, 4)
@precondition(lambda self: self.market.trades)
@rule()
def check_avg_trade_price(self):
price = sum(t.offer.price for t in self.market.trades)
energy = sum(t.offer.energy for t in self.market.trades)
assert self.market.avg_trade_price == round(price / energy, 4)
@precondition(lambda self: self.market.traded_energy)
@rule()
def check_acct(self):
actor_sums = {}
for t in self.market.trades:
actor_sums = add_or_create_key(actor_sums, t.seller,
t.offer.energy)
actor_sums = subtract_or_create_key(actor_sums, t.buyer,
t.offer.energy)
for actor, sum_ in actor_sums.items():
assert self.market.traded_energy[actor] == sum_
assert sum(self.market.traded_energy.values()) == 0
def test_market_acct_multiple(market: OneSidedMarket):
offer1 = market.offer(10, 20, 'A')
offer2 = market.offer(10, 10, 'A')
market.accept_offer(offer1, 'B')
market.accept_offer(offer2, 'C')
assert market.traded_energy['A'] == offer1.energy + offer2.energy == 30
assert market.traded_energy['B'] == -offer1.energy == -20
assert market.traded_energy['C'] == -offer2.energy == -10
assert market.bought_energy('A') == 0
assert market.sold_energy('A') == offer1.energy + offer2.energy == 30
assert market.bought_energy('B') == offer1.energy == 20
assert market.bought_energy('C') == offer2.energy == 10
def test_market_offer_invalid(market: OneSidedMarket):
with pytest.raises(InvalidOffer):
market.offer(10, -1, 'someone', 'someone')
class MarketStateMachine(RuleBasedStateMachine):
offers = Bundle('Offers')
actors = Bundle('Actors')
def __init__(self):
super().__init__()
self.market = OneSidedMarket(area=FakeArea(name='my_fake_house'))
@rule(target=actors, actor=st.text(min_size=1, max_size=3,
alphabet=string.ascii_letters + string.digits))
def new_actor(self, actor):
return actor
@rule(target=offers, seller=actors, energy=st.integers(min_value=1), price=st.integers())
def offer(self, seller, energy, price):
return self.market.offer(price, energy, seller)
@rule(offer=offers, buyer=actors)
def trade(self, offer, buyer):
assume(offer.id in self.market.offers)
self.market.accept_offer(offer, buyer)
@precondition(lambda self: self.market.offers)
@rule()
def check_avg_offer_price(self):
price = sum(o.price for o in self.market.offers.values())
energy = sum(o.energy for o in self.market.offers.values())
assert self.market.avg_offer_price == round(price / energy, 4)
@precondition(lambda self: self.market.trades)
@rule()
def check_avg_trade_price(self):
price = sum(t.offer.price for t in self.market.trades)
energy = sum(t.offer.energy for t in self.market.trades)
assert self.market.avg_trade_price == round(price / energy, 4)
@precondition(lambda self: self.market.traded_energy)
@rule()
def check_acct(self):
actor_sums = defaultdict(int)
for t in self.market.trades:
actor_sums[t.seller] += t.offer.energy
actor_sums[t.buyer] -= t.offer.energy
for actor, sum_ in actor_sums.items():
assert self.market.traded_energy[actor] == sum_
assert sum(self.market.traded_energy.values()) == 0
@precondition(lambda self: self.market.traded_energy)
@rule()
def check_iou_balance(self):
seller_ious = defaultdict(int)
buyer_ious = defaultdict(int)
for t in self.market.trades:
seller_ious[t.seller] += t.offer.price
buyer_ious[t.buyer] += t.offer.price
trade_sum = sum(t.offer.price for t in self.market.trades)
for seller, iou in seller_ious.items():
assert iou == sum(ious[seller] for ious in self.market.ious.values())
for buyer, iou in buyer_ious.items():
assert iou == sum(self.market.ious[buyer].values())
assert trade_sum == sum(sum(i.values()) for i in self.market.ious.values())
def test_market_iou(market: OneSidedMarket):
offer = market.offer(10, 20, 'A')
market.accept_offer(offer, 'B')
assert market.ious['B']['A'] == 10