class TestConfigManager(unittest.TestCase):
def setUp(self):
self.src = TestSource('test', x=5)
self.mgr = ConfigManager('prod', self.src)
def test_get_version(self):
vobj = self.mgr['2.0']
assert isinstance(vobj, ConfigVersion)
assert vobj.version == 4
vobj2 = self.mgr.get_version(pydt.datetime(2012, 2, 1))
assert vobj.tag == '2.0'
assert vobj is vobj2
vobj = self.mgr.get_version(2)
assert vobj.tag is None
assert vobj.version == 2
def test_get_config(self):
name = 'us_largecap'
date = pydt.datetime(2012, 1, 31)
vdate = pydt.datetime(2012, 1, 29)
item = self.mgr.get_config(name, 'quantmodel', date) # 2012-1-31
assert isinstance(item, ConfigItem)
assert item.universe == 'SPX'
assert item.key == 4
item_v0 = self.mgr.get_config(name, 'quantmodel', date, '1.0')
assert isinstance(item_v0, ConfigItem)
assert item_v0.key == 0
item_v8 = self.mgr.get_config(name, 'quantmodel', date, 8)
assert isinstance(item_v8, ConfigItem)
assert item_v8.key == 8
item_v4 = self.mgr.get_config(name, 'quantmodel', date, vdate)
assert isinstance(item_v4, ConfigItem)
assert item_v4.key == 4
