def test_AssetTypeDisplay(self):
asset = generate_asset()
bond = generate_bond()
self.assertEqual(asset.asset_type_display, 'Asset')
self.assertEqual(bond.asset_type_display, 'Bond Government')
self.assertEqual(asset.pricing_method(), 'Derived')
self.assertEqual(bond.pricing_method(), 'Derived')
def test_CreateMany(self):
assets = [
generate_asset(asset_manager_id=self.asset_manager_id)
for _ in range(random.randint(1, 5))
]
results = self.assets_interface.create_many(assets)
self.assertEqual(len(assets), len(results))
def setUp(self):
self.longMessage = True # Print complete error message on failure
self.maxDiff = None # View the complete diff when there is a mismatch in a test
self.interface = TransactionsInterface()
self.asset_manager_id = 1
self.asset = generate_asset(asset_manager_id=self.asset_manager_id,
fungible=True)
self.asset_book = generate_book(asset_manager_id=self.asset_manager_id)
self.counterparty_book = generate_book(
asset_manager_id=self.asset_manager_id)
self.transaction1 = generate_transaction(
asset_manager_id=self.asset_manager_id,
asset_id=self.asset.asset_id,
asset_book_id=self.asset_book.book_id,
counterparty_book_id=self.counterparty_book.book_id,
transaction_currency='USD',
net_affecting_charges=True,
charge_currency='USD')
self.transaction2 = generate_transaction(
asset_manager_id=self.asset_manager_id,
asset_id=self.asset.asset_id,
asset_book_id=self.asset_book.book_id,
counterparty_book_id=self.counterparty_book.book_id,
transaction_currency='USD',
net_affecting_charges=True,
charge_currency='USD')
self.setup_cache()
self.interface.new(self.transaction1)
self.interface.new(self.transaction2)
self.transaction_id1 = self.transaction1.transaction_id
self.transaction_id2 = self.transaction2.transaction_id
def test_FilenameToAssets(self):
# Generate file
filename = os.path.join(tempfile.gettempdir(), 'test.csv')
assets = [generate_asset() for i in range(5)]
objects_to_csv(objects=assets, filename=filename, clazz=Asset)
assets = csv_filename_to_objects(filename, json_handler=json_to_asset)
self.assertEqual(len(assets), 5)
self.assertEqual(type(assets[0]), Asset)
os.remove(filename)
def test_AssetsToCSV(self):
filename = os.path.join(tempfile.gettempdir(), 'test.csv')
assets = [generate_asset() for i in range(5)]
objects_to_csv(objects=assets, filename=filename, clazz=Asset)
# Read the file back out again
with open(filename, 'r') as temp_file:
data = temp_file.readlines()
self.assertEqual(len(data), 6) # 5 assets + header
os.remove(filename)
def setUp(self):
self.longMessage = True # Print complete error message on failure
self.maxDiff = None # View the complete diff when there is a mismatch in a test
self.interface = TransactionsInterface()
self.asset_manager_id = random.randint(1, 2**31-1)
self.asset = generate_asset(asset_manager_id=self.asset_manager_id, fungible=True)
self.trader_one_book = generate_book(asset_manager_id=self.asset_manager_id)
self.trader_two_book = generate_book(asset_manager_id=self.asset_manager_id)
self.wash_book = generate_book(asset_manager_id=self.asset_manager_id)
self.setup_cache()
def setUp(self):
self.longMessage = True # Print complete error message on failure
self.maxDiff = None # View the complete diff when there is a mismatch in a test
self.asset_manager_id = random.randint(1, 2**31-1)
self.asset = generate_asset(asset_manager_id=self.asset_manager_id)
self.asset_book = generate_book(asset_manager_id=self.asset_manager_id)
self.counterparty_book = generate_book(asset_manager_id=self.asset_manager_id)
self.transaction = generate_transaction(asset_manager_id=self.asset_manager_id, asset_id=self.asset.asset_id,
asset_book_id=self.asset_book.book_id,
counterparty_book_id=self.counterparty_book.book_id)
self.transaction_id = self.transaction.transaction_id
self.setup_cache()
def test_AssetsToCSVStream(self):
file_desc, temp_filepath = tempfile.mkstemp()
assets = [generate_asset() for i in range(5)]
with open(temp_filepath, 'w') as temp_file:
objects_to_csv_stream(objects=assets,
stream=temp_file,
clazz=Asset)
# Read the file back out again
with open(temp_filepath, 'r') as temp_file:
data = temp_file.readlines()
self.assertEqual(len(data), 6) # 5 assets + header
os.close(file_desc)
os.remove(temp_filepath)
def setUp(self):
self.longMessage = True # Print complete error message on failure
self.maxDiff = None # View the complete diff when there is a mismatch in a test
self.transactions_interface = TransactionsInterface(environment=STAGE)
self.assets_interface = AssetsInterface(environment=STAGE)
self.books_interface = BooksInterface(environment=STAGE)
self.asset_manager_id = 1
self.asset = generate_asset(asset_manager_id=self.asset_manager_id)
self.asset_book = generate_book(asset_manager_id=self.asset_manager_id)
self.counterparty_book = generate_book(
asset_manager_id=self.asset_manager_id)
self.transaction = generate_transaction(
asset_manager_id=self.asset_manager_id,
asset_id=self.asset.asset_id,
asset_book_id=self.asset_book.book_id,
counterparty_book_id=self.counterparty_book.book_id)
self.transaction_id = self.transaction.transaction_id
self.setup_cache()
def setUp(self):
self.longMessage = True # Print complete error message on failure
self.asset_manager_id = random.randint(1, 2**31 - 1)
self.asset = generate_asset(asset_manager_id=self.asset_manager_id)
self.asset_id = self.asset.asset_id
def test_AssetTypeDisplay(self):
asset = generate_asset()
bond = generate_bond()
self.assertEqual(asset.asset_type_display, 'Asset')
self.assertEqual(bond.asset_type_display, 'Bond Government')
def test_JsonToAsset(self):
asset = generate_asset()
json_asset = asset.to_json()
gen_asset = json_to_asset(json_asset)
self.assertEqual(gen_asset, asset)
