m@30: # -*- coding: utf-8 -*-
m@30: """
m@30: Created on Fri Sep  1 19:11:52 2017
m@30: 
m@30: @author: mariapanteli
m@30: """
m@30: 
m@30: import pytest
m@30: 
m@30: import numpy as np
m@30: from sklearn.model_selection import train_test_split
m@30: 
m@30: import scripts.classification as classification
m@30: 
m@30: 
m@30: def test_confusion_matrix():
m@30:     X = np.random.randn(100, 3)
m@30:     # create 2 classes by shifting the entries of half the samples
m@30:     X[-50:, :] = X[-50:, :] + 10
m@30:     Y = np.concatenate([np.repeat('a', 50), np.repeat('b', 50)])
m@30:     X_train, X_test, Y_train, Y_test = train_test_split(X, Y, train_size=0.6, random_state=1, stratify=Y)
m@93:     accuracy, _, _ = classification.confusion_matrix(X_train, Y_train, X_test, Y_test)
m@30:     # expect perfect accuracy for this 'easy' dataset
m@30:     assert accuracy == 1.0
m@30: