Feat/unit test (#438)

* refactor structure

Author: SAKURA-CAT

.gitignore

@@ -7,175 +7,17 @@ vue/components.d.ts
 vue/auto-imports.d.ts
 swanlog/
 swanlog.bak/
-logs/
 .eslintrc-auto-import.json
-
-# Byte-compiled / optimized / DLL files
-__pycache__/
-*.py[cod]
-*$py.class
-.swanlab/
-test/*
-!test/db/
-!test/_server.py
-!test/load.yaml
-!test/config.json
-
-# C extensions
-*.so
-
-# Distribution / packaging
-.Python
-wandb
-build/
-develop-eggs/
 dist/
-downloads/
-eggs/
-.eggs/
-lib/
-lib64/
-parts/
-sdist/
-var/
-wheels/
-share/python-wheels/
-*.egg-info/
-.installed.cfg
-*.egg
-MANIFEST
-
-# PyInstaller
-#  Usually these files are written by a python script from a template
-#  before PyInstaller builds the exe, so as to inject date/other infos into it.
-*.manifest
-*.spec
-
-# Installer logs
-pip-log.txt
-pip-delete-this-directory.txt
-
-# Unit test / coverage reports
-htmlcov/
-.tox/
-.nox/
-.coverage
-.coverage.*
-.cache
-nosetests.xml
-coverage.xml
-*.cover
-*.py,cover
-.hypothesis/
-.pytest_cache/
-cover/
-!unit
-unit/temp
-
-# Translations
-*.mo
-*.pot
-
-# Django stuff:
-*.log
-local_settings.py
-db.sqlite3
-db.sqlite3-journal
-
-# Flask stuff:
-instance/
-.webassets-cache
-
-# Scrapy stuff:
-.scrapy
-
-# Sphinx documentation
-docs/_build/
-
-# PyBuilder
-.pybuilder/
-target/
-
-# Jupyter Notebook
-.ipynb_checkpoints
-
-# IPython
-profile_default/
-ipython_config.py
 
-# pyenv
-#   For a library or package, you might want to ignore these files since the code is
-#   intended to run in multiple environments; otherwise, check them in:
-# .python-version
+# playground
+playground/
 
-# pipenv
-#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
-#   However, in case of collaboration, if having platform-specific dependencies or dependencies
-#   having no cross-platform support, pipenv may install dependencies that don't work, or not
-#   install all needed dependencies.
-#Pipfile.lock
 
-# poetry
-#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
-#   This is especially recommended for binary packages to ensure reproducibility, and is more
-#   commonly ignored for libraries.
-#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
-#poetry.lock
-
-# pdm
-#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
-#pdm.lock
-#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
-#   in version control.
-#   https://pdm.fming.dev/#use-with-ide
-.pdm.toml
-
-# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
-__pypackages__/
-
-# Celery stuff
-celerybeat-schedule
-celerybeat.pid
-
-# SageMath parsed files
-*.sage.py
-
-# Environments
-.venv
-env/
-venv/
-ENV/
-env.bak/
-venv.bak/
-
-# Spyder project settings
-.spyderproject
-.spyproject
-
-# Rope project settings
-.ropeproject
-
-# mkdocs documentation
-/site
-
-# mypy
-.mypy_cache/
-.dmypy.json
-dmypy.json
-
-# Pyre type checker
-.pyre/
-
-# pytype static type analyzer
-.pytype/
-
-# Cython debug symbols
-cython_debug/
-
-# PyCharm
-#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
-#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
-#  and can be added to the global gitignore or merged into this file.  For a more nuclear
-#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# Jetbrains
 .idea/
 .DS_Store
+
+# python
+__pycache__
+.pytest_cache/

.vscode/launch.json

@@ -25,52 +25,45 @@
       //sys.path 会加入顶层目录，影响模块导入查询路径
       "env": { "PYTHONPATH": "${workspaceFolder}" }
     },
-    // 开发调试
+    // 模拟实验开启
     {
-      "name": "开发调试",
+      "name": "开启一个实验",
       "type": "debugpy",
       "request": "launch",
-      "program": "${workspaceFolder}/test/test_create_experiment.py",
+      "program": "${workspaceFolder}/test/create_experiment.py",
       "console": "integratedTerminal",
       "justMyCode": true,
       "cwd": "${workspaceFolder}",
       //sys.path 会加入顶层目录，影响模块导入查询路径
       "env": { "PYTHONPATH": "${workspaceFolder}" }
     },
-    // 模拟实验开启
+    // python运行当前文件
     {
-      "name": "开启一个实验",
+      "name": "运行当前文件",
       "type": "debugpy",
       "request": "launch",
-      "program": "${workspaceFolder}/test/create_experiment.py",
+      "program": "${file}",
       "console": "integratedTerminal",
       "justMyCode": true,
       "cwd": "${workspaceFolder}",
       //sys.path 会加入顶层目录，影响模块导入查询路径
       "env": { "PYTHONPATH": "${workspaceFolder}" }
     },
     {
-      "name": "模拟命令行watch",
+      "name": "测试当前文件",
       "type": "debugpy",
       "request": "launch",
-      "program": "${workspaceFolder}/test/cli_test.py",
-      "console": "integratedTerminal",
-      "justMyCode": true,
-      "cwd": "${workspaceFolder}",
-      //sys.path 会加入顶层目录，影响模块导入查询路径
-      "env": { "PYTHONPATH": "${workspaceFolder}" }
+      "module": "pytest",
+      "args": ["${file}"],
+      "console": "integratedTerminal"
     },
-    // python运行当前文件
     {
-      "name": "Python运行当前文件",
+      "name": "进行所有单元测试",
       "type": "debugpy",
       "request": "launch",
-      "program": "${file}",
-      "console": "integratedTerminal",
-      "justMyCode": true,
-      "cwd": "${workspaceFolder}",
-      //sys.path 会加入顶层目录，影响模块导入查询路径
-      "env": { "PYTHONPATH": "${workspaceFolder}" }
+      "module": "pytest",
+      "args": ["test/unit"],
+      "console": "integratedTerminal"
     },
     // 打包命令
     {

.vscode/settings.json

@@ -89,16 +89,6 @@
 
   /** tailwind */
   "css.customData": [".vscode/tailwind.json"],
-  /** 不显示__pycache__文件夹 */
-  "files.exclude": {
-    "**/.git": true,
-    "**/.svn": true,
-    "**/.hg": true,
-    "**/CVS": true,
-    "**/.DS_Store": true,
-    "**/Thumbs.db": true,
-    "**/__pycache__": true
-  },
   /** python代码注释 */
   "autoDocstring.docstringFormat": "numpy",
 
@@ -109,8 +99,18 @@
     // 允许首行不是level1标题
     "MD041": false
   },
-  "vue-i18n.i18nPaths": "vue/src/i18n"
 
-  // /** SQLite */
-  // "sqlite.databaseExtensions": ["db", "db3", "sdb", "s3db", "sqlite", "sqlite3", "swanlab"]
+  /** 不显示文件夹 */
+  "files.exclude": {
+    "**/.git": true,
+    "**/.svn": true,
+    "**/.hg": true,
+    "**/CVS": true,
+    "**/.DS_Store": true,
+    "**/Thumbs.db": true,
+    "**/__pycache__": true,
+    ".idea": true
+  },
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true
 }

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+python_files = pytest_*

test/_server.py

@@ -20,8 +20,8 @@
 swanlab.init(
     description="这是一个测试实验",
     log_level="debug",
-    config="test/config.json",
-    load="test/load.yaml",
+    config="test/config/config.json",
+    load="test/config/load.yaml",
     # cloud=True,
 )
 swanlab.config.epoches = epochs

test/cli_test.py

@@ -10,7 +10,15 @@
     增加了实际开发中不会用到的热启动功能
 """
 import uvicorn
+from swanlab.db import connect
+from swanlab.server.app import app
+
+connect()
+
+
+def main():
+    return app
 
 
 if __name__ == "__main__":
-    uvicorn.run("_server:main", host="0.0.0.0", port=6092, reload=True, log_level="critical")
+    uvicorn.run("start_server:main", host="0.0.0.0", port=6092, reload=True, log_level="critical")