Skip to content

Commit

Permalink
Add tests folders
Browse files Browse the repository at this point in the history
  • Loading branch information
Matteo Serafino committed Jul 17, 2022
1 parent 5519b8b commit a03ddd2
Showing 1 changed file with 68 additions and 0 deletions.
68 changes: 68 additions & 0 deletions tests/test_dimensionality_reduction.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
import unittest
from sklearn import datasets
from src.dimensionality_reduction import DimensionalityReduction

dr = DimensionalityReduction()

class TestDimensionalityReduction(unittest.TestCase):

def setUp(self):

iris_dataset = datasets.load_iris()

self.X = iris_dataset.data[:, :3]
self.y = iris_dataset.target

def test_tsne(self):

df = dr.tsne(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_umap(self):

df = dr.UMAP(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_lda(self):

df = dr.lda(self.X, self.y)

assert df.shape == (150, 3)

def test_pca(self):

df = dr.pca(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_factor_analysis(self):

df = dr.factor_analysis(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_truncated_svd(self):

df = dr.truncated_svd(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_kernal_pca(self):

df = dr.kernel_pca(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_multidim_scaling(self):

df = dr.multidim_scaling(self.X, self.y, 2)

assert df.shape == (150, 3)

def test_isomap(self):

df = dr.isomap(self.X, self.y, 2)

assert df.shape == (150, 3)

0 comments on commit a03ddd2

Please sign in to comment.