update tests

vtuber-plan · Jul 8, 2024 · b56d0b2 · b56d0b2
1 parent f5e792f
commit b56d0b2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/.github/workflows/dev.yml b/.github/workflows/dev.yml
@@ -31,6 +31,7 @@ jobs:
           cd ${{ github.workspace }}
           pip install --upgrade pip
           pip install -e .
+          pip install -r requirements
 
       - name: Test Olah
         run: |

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -32,6 +32,7 @@ jobs:
           cd ${{ github.workspace }}
           pip install --upgrade pip
           pip install -e .
+          pip install -r requirements
 
       - name: Test Olah
         run: |

diff --git a/tests/simple_test.py b/tests/simple_test.py
@@ -4,12 +4,22 @@
 
 from huggingface_hub import snapshot_download
 
-def test_simple():
+def test_dataset():
     process = subprocess.Popen(['python', '-m', 'olah.server'])
 
     os.environ['HF_ENDPOINT'] = 'http://localhost:8090'
     snapshot_download(repo_id='Nerfgun3/bad_prompt', repo_type='dataset',
                     local_dir='./dataset_dir', max_workers=8)
 
+    # 终止子进程
+    process.terminate()
+
+def test_model():
+    process = subprocess.Popen(['python', '-m', 'olah.server'])
+
+    os.environ['HF_ENDPOINT'] = 'http://localhost:8090'
+    snapshot_download(repo_id='prajjwal1/bert-tiny', repo_type='model',
+                    local_dir='./model_dir', max_workers=8)
+
     # 终止子进程
     process.terminate()