def test_call_super_test():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
MSLR30K(tmpdir, split="test", fold=5)
mock_super.called_once()
args, kwargs = mock_super.call_args
assert kwargs["file"] == os.path.join(tmpdir, "Fold5", "test.txt")
assert kwargs["normalize"]
assert kwargs["filter_queries"]
def test_call_validate_download():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
MSLR30K(tmpdir, split="train")
mock_vali.called_once()
args, kwargs = mock_vali.call_args
assert kwargs["location"] == tmpdir
assert kwargs["validate_checksums"]
assert isinstance(kwargs["expected_files"], list)
def test_call_super_vali():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
IstellaS(tmpdir, split="vali")
mock_super.called_once()
args, kwargs = mock_super.call_args
assert kwargs["file"] == os.path.join(tmpdir, "sample", "vali.txt")
assert kwargs["normalize"]
assert kwargs["filter_queries"]
def test_call_super_train():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
Istella(tmpdir, split="train")
mock_super.called_once()
args, kwargs = mock_super.call_args
assert kwargs["file"] == os.path.join(tmpdir, "full", "train.txt")
assert kwargs["normalize"]
assert not kwargs["filter_queries"]
def test_wrong_split_raises_error():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
with pytest.raises(ValueError):
MSLR30K(tmpdir, split="nonexisting")
def test_wrong_fold_raises_error():
with mock_svmrank_dataset(pkg) as (tmpdir, mock_super, mock_vali):
with pytest.raises(ValueError):
MSLR30K(tmpdir, split="train", fold=99)
