Added unit tests for SVM (#1)

Makefile

@@ -16,4 +16,7 @@ local-install:
 	$(ENV_PATH)pip install .
 
 test:
+	$(ENV_PATH)pytest -m "not slow"
+
+test-full:
 	$(ENV_PATH)pytest

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+markers = slow: marks tests as slow (deselect with '-m "not slow"')

tests/test_discovery_svm.py

@@ -57,4 +57,20 @@ def test_preprocess(mock_read_csv, mock_data):
     X, y = discovery_svm.extract_features(data)
     assert X.shape == (100, 44)
 
-# TODO: add tests for SVM grid_search and evaluate_model
+@pytest.mark.slow
+def test_grid_search():
+    X = np.random.randn(100, 44)
+    y = np.random.choice([0, 1], 100)
+    grid = discovery_svm.grid_search(X, y)
+    assert isinstance(grid, discovery_svm.GridSearchCV)
+    assert hasattr(grid, 'best_params_')
+    assert hasattr(grid, 'best_score_')
+
+def test_evaluate_model():
+    X = np.random.randn(100, 44)
+    y = np.random.choice([0, 1], 100)
+    stats = discovery_svm.evaluate_model(X, y)
+    assert isinstance(stats, tuple)
+    assert len(stats) == 3
+    for i in range(3):
+        assert len(stats[i]) == 5