Spaces:
Sleeping
Sleeping
File size: 1,160 Bytes
be7c937 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | """Pytest configuration and fixtures for WayyDB tests."""
import pytest
import numpy as np
import tempfile
import shutil
from pathlib import Path
@pytest.fixture
def temp_dir():
"""Create a temporary directory for test data."""
path = tempfile.mkdtemp(prefix="wayy_test_")
yield path
shutil.rmtree(path, ignore_errors=True)
@pytest.fixture
def sample_trades():
"""Sample trades data for testing."""
return {
"timestamp": np.array([1000, 2000, 3000, 4000, 5000], dtype=np.int64),
"symbol": np.array([0, 1, 0, 1, 0], dtype=np.uint32), # AAPL, MSFT alternating
"price": np.array([150.0, 380.0, 151.0, 381.0, 152.0], dtype=np.float64),
"size": np.array([100, 200, 150, 250, 100], dtype=np.int64),
}
@pytest.fixture
def sample_quotes():
"""Sample quotes data for testing."""
return {
"timestamp": np.array([500, 900, 1500, 2500, 3500], dtype=np.int64),
"symbol": np.array([0, 1, 0, 1, 0], dtype=np.uint32),
"bid": np.array([149.5, 379.5, 150.5, 380.5, 151.5], dtype=np.float64),
"ask": np.array([150.0, 380.0, 151.0, 381.0, 152.0], dtype=np.float64),
}
|