mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-31 01:17:42 +00:00
you-n-g
943d2087fc
* refactor: Replace ExtendedSettingsConfigDict with SettingsConfigDict * lint * lint
19 lines
492 B
Python
19 lines
492 B
Python
import unittest
|
|
|
|
|
|
class ConfUtils(unittest.TestCase):
|
|
|
|
def test_conf(self):
|
|
import os
|
|
|
|
from rdagent.utils.env import 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
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|