test_pay.py

import address
import coin
import pay
from dumb25519 import random_point
import unittest

class TestPay(unittest.TestCase):
	def test_complete(self):
		value_bytes = 4
		memo_bytes = 16
		address_params = address.AddressParameters(random_point(),random_point())
		coin_params = coin.CoinParameters(address_params.F,address_params.G,random_point(),random_point(),value_bytes,memo_bytes)
		pay_params = pay.PayParameters(address_params.F,address_params.G,coin_params.H,coin_params.value_bytes)

		public = address.SpendKey(address_params).public_address()
		coin_ = coin.Coin(coin_params,public,1,'Test memo',False,True)

		witness = pay.PayWitness(coin_.k)
		statement = pay.PayStatement(pay_params,'Proof context',coin_,coin_.k*public.Q1,coin_.k*address_params.F,public)

		proof = pay.prove(statement,witness)
		pay.verify(statement,proof)

		with self.assertRaises(ArithmeticError):
			statement.context = 'Evil context'
			pay.verify(statement,proof)

if __name__ == '__main__':
	unittest.main()