def test_insert_metadata(self): finder = AssetFinder() finder.insert_metadata( 0, asset_type='equity', start_date='2014-01-01', end_date='2015-01-01', symbol="PLAY", foo_data="FOO", ) # Test proper insertion self.assertEqual('equity', finder.metadata_cache[0]['asset_type']) self.assertEqual('PLAY', finder.metadata_cache[0]['symbol']) self.assertEqual('2015-01-01', finder.metadata_cache[0]['end_date']) # Test invalid field self.assertFalse('foo_data' in finder.metadata_cache[0]) # Test updating fields finder.insert_metadata( 0, asset_type='equity', start_date='2014-01-01', end_date='2015-02-01', symbol="PLAY", exchange="NYSE", ) self.assertEqual('2015-02-01', finder.metadata_cache[0]['end_date']) self.assertEqual('NYSE', finder.metadata_cache[0]['exchange']) # Check that old data survived self.assertEqual('PLAY', finder.metadata_cache[0]['symbol'])
def test_insert_metadata(self): finder = AssetFinder() finder.insert_metadata(0, asset_type='equity', start_date='2014-01-01', end_date='2015-01-01', symbol="PLAY", foo_data="FOO",) # Test proper insertion self.assertEqual('equity', finder.metadata_cache[0]['asset_type']) self.assertEqual('PLAY', finder.metadata_cache[0]['symbol']) self.assertEqual('2015-01-01', finder.metadata_cache[0]['end_date']) # Test invalid field self.assertFalse('foo_data' in finder.metadata_cache[0]) # Test updating fields finder.insert_metadata(0, asset_type='equity', start_date='2014-01-01', end_date='2015-02-01', symbol="PLAY", exchange="NYSE",) self.assertEqual('2015-02-01', finder.metadata_cache[0]['end_date']) self.assertEqual('NYSE', finder.metadata_cache[0]['exchange']) # Check that old data survived self.assertEqual('PLAY', finder.metadata_cache[0]['symbol'])
def test_insert_metadata(self): finder = AssetFinder() finder.insert_metadata(0, asset_type='equity', start_date='2014-01-01', end_date='2015-01-01', symbol="PLAY", foo_data="FOO",) # Test proper insertion equity = finder.retrieve_asset(0) self.assertIsInstance(equity, Equity) self.assertEqual('PLAY', equity.symbol) self.assertEqual(pd.Timestamp('2015-01-01', tz='UTC'), equity.end_date) # Test invalid field self.assertFalse('foo_data' in finder.metadata_cache[0])
def test_insert_metadata(self): finder = AssetFinder() finder.insert_metadata( 0, asset_type='equity', start_date='2014-01-01', end_date='2015-01-01', symbol="PLAY", foo_data="FOO", ) # Test proper insertion equity = finder.retrieve_asset(0) self.assertIsInstance(equity, Equity) self.assertEqual('PLAY', equity.symbol) self.assertEqual(pd.Timestamp('2015-01-01', tz='UTC'), equity.end_date) # Test invalid field self.assertFalse('foo_data' in finder.metadata_cache[0])