def test_transferOCEAN(): w1 = AgentWallet(OCEAN=10.0) w2 = AgentWallet(OCEAN=1.0) w1.transferOCEAN(w2, 2.0) assert w1.OCEAN() == (10.0 - 2.0) assert w2.OCEAN() == (1.0 + 2.0)
def test_transferUSD(): w1 = AgentWallet(USD=10.0) w2 = AgentWallet(USD=1.0) w1.transferUSD(w2, 2.0) assert w1.USD() == 10.0 - 2.0 assert w2.USD() == 1.0 + 2.0
def testUSD(): w = AgentWallet() w.depositUSD(13.25) assert w.USD() == 13.25 w.depositUSD(1.00) assert w.USD() == 14.25 w.withdrawUSD(2.10) assert w.USD() == 12.15 w.depositUSD(0.0) w.withdrawUSD(0.0) assert w.USD() == 12.15 assert w.totalUSDin() == (13.25 + 1.0) with pytest.raises(AssertionError): w.depositUSD(-5.0) with pytest.raises(AssertionError): w.withdrawUSD(-5.0) with pytest.raises(ValueError): w.withdrawUSD(1000.0)
def __init__(self, name: str, USD: float, OCEAN: float): self.name = name self._wallet = AgentWallet.AgentWallet(USD, OCEAN) #postconditions assert self.USD() == USD assert self.OCEAN() == OCEAN
def testOCEAN(): w = AgentWallet() w.depositOCEAN(13.25) assert w.OCEAN() == 13.25 w.depositOCEAN(1.00) assert w.OCEAN() == 14.25 w.withdrawOCEAN(2.10) assert w.OCEAN() == 12.15 w.depositOCEAN(0.0) w.withdrawOCEAN(0.0) assert w.OCEAN() == 12.15 assert w.totalOCEANin() == (13.25 + 1.0) with pytest.raises(AssertionError): w.depositOCEAN(-5.0) with pytest.raises(AssertionError): w.withdrawOCEAN(-5.0) with pytest.raises(ValueError): w.withdrawOCEAN(1000.0)
def test_initFromPrivateKey(): private_key = '0xbbfbee4961061d506ffbb11dfea64eba16355cbf1d9c29613126ba7fec0aed5d' address = '0x66aB6D9362d4F35596279692F0251Db635165871' w = AgentWallet(private_key=private_key) assert w._address == address assert w._web3wallet.address == address assert w._web3wallet.private_key == private_key
def _make_info(private_key_name: str): class _Info: pass info = _Info() network = web3util.get_network() info.private_key = web3util.confFileValue(network, private_key_name) info.agent_wallet = AgentWallet.AgentWallet(OCEAN=_OCEAN_INIT, private_key=info.private_key) info.web3wallet = info.agent_wallet._web3wallet info.DT = _createDT(info.web3wallet) info.pool = _createPool(DT=info.DT, web3_w=info.web3wallet) return info
def _make_wallet(private_key_name:str, cache: bool): global _CACHED_WALLET if _CACHED_WALLET is None and cache: class _Wallet: pass wallet = _Wallet() network = web3util.get_network() wallet.private_key = web3util.confFileValue(network, private_key_name) wallet.agent_wallet = AgentWallet.AgentWallet( OCEAN=_OCEAN_INIT,private_key=wallet.private_key) wallet.web3wallet = wallet.agent_wallet._web3wallet wallet.DT = _createDT(wallet.web3wallet) wallet.pool = _createPool(DT=wallet.DT, web3_w=wallet.web3wallet) _CACHED_WALLET = wallet return _CACHED_WALLET return _CACHED_WALLET
def __init__(self, name: str, USD: float, OCEAN: float, private_key: str = None): self.name = name self._wallet = AgentWallet.AgentWallet(USD, OCEAN, private_key=private_key) #postconditions if private_key == None: assert self.USD() == USD assert self.OCEAN() == OCEAN else: # TODO # This is kind of weak - we should test for sum of current and previous assert self.USD() >= USD assert self.OCEAN() >= OCEAN
def _make_info(private_key_name: str): class _Info: def __init__(self): self.private_key: Union[str, None] = None self.agent_wallet: Union[AgentWallet.AgentWallet, None] = None self.web3wallet: Union[web3wallet, None] = None self.DT: Union[datatoken, None] = None self.pool: Union[bool, None] = None info = _Info() network = web3util.get_network() info.private_key = web3util.confFileValue(network, private_key_name) info.agent_wallet = AgentWallet.AgentWallet(OCEAN=_OCEAN_INIT, private_key=info.private_key) info.web3wallet = info.agent_wallet._web3wallet info.DT = _createDT(info.web3wallet) info.pool = _createPool(DT=info.DT, web3_w=info.web3wallet) return info
def test_initRandomPrivateKey(): w1 = AgentWallet() w2 = AgentWallet() assert w1._address != w2._address assert w1._web3wallet.private_key != w2._web3wallet.private_key
def testETH1(): #super-basic test for ETH w = AgentWallet() assert isinstance(w.ETH(), float)
def testFloatingPointRoundoff_USD(): w = AgentWallet(USD=2.4) w.withdrawUSD(2.4000000000000004) #should not get ValueError assert w.USD() == 0.0 assert w.OCEAN() == 0.0
def testETH2(): #TEST_PRIVATE_KEY1 should get initialized with ETH when ganache starts network = web3util.get_network() private_key = web3util.confFileValue(network, 'TEST_PRIVATE_KEY1') w = AgentWallet(private_key=private_key) assert w.ETH() > 1.0
def testInitiallyFilled(): w = AgentWallet(USD=1.2, OCEAN=3.4) assert w.USD() == 1.2 assert w.OCEAN() == 3.4
def testInitiallyEmpty(): w = AgentWallet() assert w.USD() == 0.0 assert w.OCEAN() == 0.0
def test_gotSomeETHforGas(): w = AgentWallet() assert w.ETH() > 0.0
def testStr(): w = AgentWallet() assert "AgentWallet" in str(w)