mirror of
https://github.com/OpenSPG/openspg.git
synced 2025-07-03 15:15:42 +00:00
61 lines
1.7 KiB
Python
Executable File
61 lines
1.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
# -*- mode: python -*-
|
|
|
|
# Copyright 2023 OpenSPG Authors
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
|
|
# in compliance with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software distributed under the License
|
|
# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
# or implied.
|
|
|
|
from __future__ import print_function
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
|
|
|
|
class NN4KTestsRunner(object):
|
|
def _run_all_tests(self):
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
dir_path = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
restore_script_path = os.path.join(dir_path, "python-env", ".env.restore.sh")
|
|
args = [restore_script_path]
|
|
subprocess.check_call(args)
|
|
|
|
nn4k_dir_path = os.path.dirname(dir_path)
|
|
python_executable_path = os.path.join(
|
|
dir_path, "python-env", ".env", "bin", "python"
|
|
)
|
|
saved_dir_path = os.getcwd()
|
|
os.chdir(dir_path)
|
|
|
|
args = ["env", "PYTHONPATH=%s" % nn4k_dir_path]
|
|
args += [python_executable_path]
|
|
args += ["-m", "unittest"]
|
|
try:
|
|
subprocess.check_call(args)
|
|
except subprocess.CalledProcessError:
|
|
raise SystemExit(1)
|
|
finally:
|
|
os.chdir(saved_dir_path)
|
|
|
|
def run(self):
|
|
self._run_all_tests()
|
|
|
|
|
|
def main():
|
|
runner = NN4KTestsRunner()
|
|
runner.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|