-
-
Notifications
You must be signed in to change notification settings - Fork 1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1655 from VesnaT/flagged_data
[ENH] Scatterplot, HeatMap, TreeGraph, ConfusionMatrix and Unsupervised widgets: Output Flagged Data
- Loading branch information
Showing
18 changed files
with
554 additions
and
127 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
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
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
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
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,102 @@ | ||
| import random | ||
| import unittest | ||
|
|
||
| import numpy as np | ||
|
|
||
| from Orange.data import Table, Variable | ||
| from Orange.widgets.utils.annotated_data import (create_annotated_table, | ||
| ANNOTATED_DATA_FEATURE_NAME) | ||
|
|
||
|
|
||
| class TestAnnotatedData(unittest.TestCase): | ||
| def setUp(self): | ||
| Variable._clear_all_caches() | ||
| random.seed(42) | ||
| self.zoo = Table("zoo") | ||
|
|
||
| def test_create_annotated_table(self): | ||
| annotated = create_annotated_table(self.zoo, list(range(10))) | ||
|
|
||
| # check annotated table domain | ||
| self.assertEqual(annotated.domain.variables, self.zoo.domain.variables) | ||
| self.assertEqual(2, len(annotated.domain.metas)) | ||
| self.assertIn(self.zoo.domain.metas[0], annotated.domain.metas) | ||
| self.assertIn(ANNOTATED_DATA_FEATURE_NAME, | ||
| [m.name for m in annotated.domain.metas]) | ||
|
|
||
| # check annotated table data | ||
| np.testing.assert_array_equal(annotated.X, self.zoo.X) | ||
| np.testing.assert_array_equal(annotated.Y, self.zoo.Y) | ||
| np.testing.assert_array_equal(annotated.metas[:, 0].ravel(), | ||
| self.zoo.metas.ravel()) | ||
| self.assertEqual( | ||
| 10, np.sum([i[ANNOTATED_DATA_FEATURE_NAME] for i in annotated])) | ||
|
|
||
| def test_create_annotated_table_selected(self): | ||
| # check annotated column for no selected indices | ||
| annotated = create_annotated_table(self.zoo, []) | ||
| self.assertEqual(len(annotated), len(self.zoo)) | ||
| self.assertEqual( | ||
| 0, np.sum([i[ANNOTATED_DATA_FEATURE_NAME] for i in annotated])) | ||
|
|
||
| # check annotated column fol all selectes indices | ||
| annotated = create_annotated_table(self.zoo, list(range(len(self.zoo)))) | ||
| self.assertEqual(len(annotated), len(self.zoo)) | ||
| self.assertEqual( | ||
| len(self.zoo), | ||
| np.sum([i[ANNOTATED_DATA_FEATURE_NAME] for i in annotated])) | ||
|
|
||
| def test_create_annotated_table_none_data(self): | ||
| self.assertIsNone(create_annotated_table(None, None)) | ||
|
|
||
| def test_create_annotated_table_none_indices(self): | ||
| annotated = create_annotated_table(self.zoo, None) | ||
| self.assertEqual(len(annotated), len(self.zoo)) | ||
| self.assertEqual( | ||
| 0, np.sum([i[ANNOTATED_DATA_FEATURE_NAME] for i in annotated])) | ||
|
|
||
| def test_cascade_annotated_tables(self): | ||
| # check cascade of annotated tables | ||
| data = self.zoo | ||
| data.domain.metas[0].name = ANNOTATED_DATA_FEATURE_NAME | ||
| for i in range(5): | ||
| data = create_annotated_table( | ||
| data, random.sample(range(0, len(self.zoo)), 20)) | ||
| self.assertEqual(2 + i, len(data.domain.metas)) | ||
| self.assertIn(self.zoo.domain.metas[0], data.domain.metas) | ||
| self.assertIn(ANNOTATED_DATA_FEATURE_NAME, | ||
| [m.name for m in data.domain.metas]) | ||
| for j in range(2, i + 3): | ||
| self.assertIn("{} ({})".format(ANNOTATED_DATA_FEATURE_NAME, j), | ||
| [m.name for m in data.domain.metas]) | ||
|
|
||
| def test_cascade_annotated_tables_with_missing_middle_feature(self): | ||
| # check table for domain [..., "Feature", "Selected", "Selected (3)] -> | ||
| # [..., "Feature", "Selected", "Selected (3), "Selected (4)"] | ||
| data = self.zoo | ||
| data.domain.attributes[0].name = ANNOTATED_DATA_FEATURE_NAME | ||
| data.domain.metas[0].name = "{} ({})".format( | ||
| ANNOTATED_DATA_FEATURE_NAME, 3) | ||
| data = create_annotated_table( | ||
| data, random.sample(range(0, len(self.zoo)), 20)) | ||
| self.assertEqual(2, len(data.domain.metas)) | ||
| self.assertEqual(data.domain.attributes[0].name, | ||
| ANNOTATED_DATA_FEATURE_NAME) | ||
| self.assertEqual(data.domain.metas[0].name, | ||
| "{} ({})".format(ANNOTATED_DATA_FEATURE_NAME, 3)) | ||
| self.assertEqual(data.domain.metas[1].name, | ||
| "{} ({})".format(ANNOTATED_DATA_FEATURE_NAME, 4)) | ||
|
|
||
| def test_cascade_annotated_tables_with_missing_annotated_feature(self): | ||
| # check table for domain [..., "Feature", "Selected (3)] -> | ||
| # [..., "Feature", "Selected (3), "Selected (4)"] | ||
| data = self.zoo | ||
| data.domain.metas[0].name = "{} ({})".format( | ||
| ANNOTATED_DATA_FEATURE_NAME, 3) | ||
| data = create_annotated_table( | ||
| data, random.sample(range(0, len(self.zoo)), 20)) | ||
| self.assertEqual(2, len(data.domain.metas)) | ||
| self.assertEqual(data.domain.metas[0].name, | ||
| "{} ({})".format(ANNOTATED_DATA_FEATURE_NAME, 3)) | ||
| self.assertEqual(data.domain.metas[1].name, | ||
| "{} ({})".format(ANNOTATED_DATA_FEATURE_NAME, 4)) |
Oops, something went wrong.