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)