diff --git a/alfie/tests/__pycache__/test_training.cpython-37-pytest-5.2.1.pyc b/alfie/tests/__pycache__/test_training.cpython-37-pytest-5.2.1.pyc index 9d51d42..080d758 100644 Binary files a/alfie/tests/__pycache__/test_training.cpython-37-pytest-5.2.1.pyc and b/alfie/tests/__pycache__/test_training.cpython-37-pytest-5.2.1.pyc differ diff --git a/alfie/tests/test_training.py b/alfie/tests/test_training.py index 6977234..f2401a9 100644 --- a/alfie/tests/test_training.py +++ b/alfie/tests/test_training.py @@ -24,6 +24,10 @@ def test_split(): assert test.shape == (5, 2) assert list(test.index) == [15, 21, 7, 11, 8] + #resplit to test the print line + train, test = training.stratified_taxon_split(data, class_col = "phylum", + test_size = .2, silent = True, seed = 1738) + def test_sample_seq(): @@ -40,6 +44,9 @@ def test_sample_seq(): assert out2 == expected2 + with pytest.raises(ValueError): + training.sample_seq("ATGA", min_size = 100) + def test_process_sequences():