-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Matteo Serafino
committed
Jul 17, 2022
1 parent
5519b8b
commit a03ddd2
Showing
1 changed file
with
68 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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) |