mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
you-n-g
29 lines
846 B
Python
29 lines
846 B
Python
import unittest
|
|
|
|
|
|
class ConfUtils(unittest.TestCase):
|
|
|
|
def test_conf(self):
|
|
import os
|
|
|
|
from rdagent.utils.env import EnvConf, QlibDockerConf
|
|
|
|
os.environ["MEM_LIMIT"] = "200g"
|
|
assert QlibDockerConf().mem_limit == "200g" # base class will affect subclasses
|
|
os.environ["QLIB_DOCKER_MEM_LIMIT"] = "300g"
|
|
assert QlibDockerConf().mem_limit == "300g" # more accurate subclass will override the base class
|
|
|
|
os.environ["default_entry"] = "which python"
|
|
os.environ["ENABLE_CACHE"] = "False"
|
|
|
|
assert EnvConf().enable_cache is False
|
|
assert QlibDockerConf().enable_cache is False
|
|
|
|
os.environ["ENABLE_CACHE"] = "True"
|
|
assert EnvConf().enable_cache is True
|
|
assert QlibDockerConf().enable_cache is True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|