final training lines tests

CNuge · Mar 11, 2020 · e62f55c · e62f55c
1 parent a378345
commit e62f55c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
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
diff --git 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():