configuration: added 'glossaries' directory
This commit is contained in:
@@ -71,11 +71,13 @@ class TestConfig(unittest.TestCase):
|
||||
'frequency_penalty': 0.7,
|
||||
'presence_penalty': 0.2
|
||||
}
|
||||
}
|
||||
},
|
||||
'glossaries': './glossaries/'
|
||||
}
|
||||
config = Config.from_dict(source_dict)
|
||||
self.assertEqual(config.cache, '.')
|
||||
self.assertEqual(config.db, './test_db/')
|
||||
self.assertEqual(config.glossaries, './glossaries/')
|
||||
self.assertEqual(len(config.ais), 1)
|
||||
self.assertEqual(config.ais['myopenai'].name, 'openai')
|
||||
self.assertEqual(cast(OpenAIConfig, config.ais['myopenai']).system, 'Custom system')
|
||||
@@ -105,6 +107,7 @@ class TestConfig(unittest.TestCase):
|
||||
'frequency_penalty': 0.7,
|
||||
'presence_penalty': 0.2
|
||||
}
|
||||
# omit glossaries, since it's optional
|
||||
}
|
||||
}
|
||||
with open(self.test_file.name, 'w') as f:
|
||||
@@ -113,6 +116,8 @@ class TestConfig(unittest.TestCase):
|
||||
self.assertIsInstance(config, Config)
|
||||
self.assertEqual(config.cache, './test_cache/')
|
||||
self.assertEqual(config.db, './test_db/')
|
||||
# missing 'glossaries' should result in 'None'
|
||||
self.assertEqual(config.glossaries, None)
|
||||
self.assertEqual(len(config.ais), 1)
|
||||
self.assertIsInstance(config.ais['default'], AIConfig)
|
||||
self.assertEqual(cast(OpenAIConfig, config.ais['default']).system, 'Custom system')
|
||||
|
||||
Reference in New Issue
Block a user