Skip to content
Snippets Groups Projects
Commit 023bb810 authored by Quoc-Tuan Truong's avatar Quoc-Tuan Truong
Browse files

Add tests for base_strategy and ratio_split

parent 58e2afaa
No related branches found
No related tags found
No related merge requests found
# -*- coding: utf-8 -*-
"""
@author: Quoc-Tuan Truong <tuantq.vnu@gmail.com>
"""
from ..base_strategy import BaseStrategy
def test_init():
bs = BaseStrategy(None, verbose=True)
assert not bs.exclude_unknowns
assert 1. == bs.rating_threshold
def test_trainset_none():
bs = BaseStrategy(None, verbose=True)
try:
bs.evaluate(None, {}, False)
except ValueError:
assert True
def test_testset_none():
bs = BaseStrategy(None, train_set=[], verbose=True)
try:
bs.evaluate(None, {}, False)
except ValueError:
assert True
# -*- coding: utf-8 -*-
"""
@author: Quoc-Tuan Truong <tuantq.vnu@gmail.com>
"""
from ..ratio_split import RatioSplit
def test_validate_size():
train_size, val_size, test_size = RatioSplit._validate_sizes(0.1, 0.2, 10)
assert 7 == train_size
assert 1 == val_size
assert 2 == test_size
train_size, val_size, test_size = RatioSplit._validate_sizes(None, 0.5, 10)
assert 5 == train_size
assert 0 == val_size
assert 5 == test_size
train_size, val_size, test_size = RatioSplit._validate_sizes(None, None, 10)
assert 10 == train_size
assert 0 == val_size
assert 0 == test_size
train_size, val_size, test_size = RatioSplit._validate_sizes(2, 2, 10)
assert 6 == train_size
assert 2 == val_size
assert 2 == test_size
try:
RatioSplit._validate_sizes(-1, 0.2, 10)
except ValueError:
assert True
try:
RatioSplit._validate_sizes(11, 0.2, 10)
except ValueError:
assert True
try:
RatioSplit._validate_sizes(0, 11, 10)
except ValueError:
assert True
try:
RatioSplit._validate_sizes(3, 8, 10)
except ValueError:
assert True
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment