Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
feat: add unittests
Browse files
tests/src/test_benchmarks.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from src.benchmarks import BenchmarksQA, BenchmarksLongDoc
|
| 2 |
+
|
| 3 |
+
|
| 4 |
+
def test_qabenchmarks():
|
| 5 |
+
print(list(BenchmarksQA))
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
def test_longdocbenchmarks():
|
| 9 |
+
print(list(BenchmarksLongDoc))
|