17 lines
382 B
Python
17 lines
382 B
Python
import pytest
|
|
|
|
from pygrex.config import cfg
|
|
from pygrex.data_reader.data_reader import DataReader
|
|
|
|
|
|
@pytest.fixture
|
|
def data_reader():
|
|
return DataReader(**cfg.data.testdata)
|
|
|
|
|
|
def test_import(data_reader):
|
|
assert data_reader.num_user == 249
|
|
assert data_reader.num_item == 551
|
|
assert data_reader.dataset.shape[0] == 1000
|
|
assert data_reader.dataset.shape[1] == 4
|