From 2d166bd00595d9f4c319aa5286587aac20e5627e Mon Sep 17 00:00:00 2001 From: Matthieu Hog Date: Fri, 26 Jul 2024 16:36:00 +0200 Subject: [PATCH] added loggin test and command line node --- tests/nodes/plugins/dummyNodes.py | 82 ++++++++++++++++++++++++++++--- 1 file changed, 76 insertions(+), 6 deletions(-) diff --git a/tests/nodes/plugins/dummyNodes.py b/tests/nodes/plugins/dummyNodes.py index 02622992e5..d5ded2e06c 100644 --- a/tests/nodes/plugins/dummyNodes.py +++ b/tests/nodes/plugins/dummyNodes.py @@ -1,7 +1,9 @@ import os -from meshroom.core.plugin import PluginNode, EnvType +from meshroom.core.plugin import PluginNode, PluginCommandLineNode, EnvType + +#Python nodes class DummyConda(PluginNode): @@ -16,7 +18,7 @@ class DummyConda(PluginNode): def processChunk(self, chunk): import numpy as np - print(np.abs(-1)) + chunk.logger.info(np.abs(-1)) class DummyDocker(PluginNode): @@ -31,7 +33,7 @@ class DummyDocker(PluginNode): def processChunk(self, chunk): import numpy as np - print(np.abs(-1)) + chunk.logger.info(np.abs(-1)) class DummyVenv(PluginNode): @@ -47,7 +49,7 @@ class DummyVenv(PluginNode): def processChunk(self, chunk): import numpy as np - print(np.abs(-1)) + chunk.logger.info(np.abs(-1)) class DummyPip(PluginNode): @@ -62,7 +64,7 @@ class DummyPip(PluginNode): def processChunk(self, chunk): import numpy as np - print(np.abs(-1)) + chunk.logger.info(np.abs(-1)) class DummyNone(PluginNode): @@ -77,4 +79,72 @@ class DummyNone(PluginNode): def processChunk(self, chunk): import numpy as np - print(np.abs(-1)) \ No newline at end of file + chunk.logger.info(np.abs(-1)) + +#Command line node + +class DummyCondaCL(PluginCommandLineNode): + + category = 'Dummy' + documentation = ''' ''' + + envType = EnvType.CONDA + envFile = os.path.join(os.path.dirname(__file__), "env.yaml") + + inputs = [] + outputs = [] + + commandLine = "python -c \"import numpy as np; import logging; logging.info(np.abs(-1))\"" + +class DummyDockerCL(PluginCommandLineNode): + + category = 'Dummy' + documentation = ''' ''' + + envType = EnvType.DOCKER + envFile = os.path.join(os.path.dirname(__file__), "Dockerfile") + + inputs = [] + outputs = [] + + commandLine = "python -c \"import numpy as np; import logging; logging.info(np.abs(-1))\"" + + +class DummyVenvCL(PluginCommandLineNode): + + category = 'Dummy' + documentation = ''' ''' + + envType = EnvType.VENV + envFile = os.path.join(os.path.dirname(__file__), "requirements.txt") + + inputs = [] + outputs = [] + + commandLine = "python -c \"import numpy as np; import logging; logging.info(np.abs(-1))\"" + +class DummyPipCL(PluginCommandLineNode): + + category = 'Dummy' + documentation = ''' ''' + + envType = EnvType.PIP + envFile = os.path.join(os.path.dirname(__file__), "requirements.txt") + + inputs = [] + outputs = [] + + commandLine = "python -c \"import numpy as np; import logging; logging.info(np.abs(-1))\"" + +class DummyNoneCL(PluginCommandLineNode): + + category = 'Dummy' + documentation = ''' ''' + + envType = EnvType.NONE + envFile = None + + inputs = [] + outputs = [] + + commandLine = "python -c \"import numpy as np; import logging; logging.info(np.abs(-1))\"" \ No newline at end of file