| @@ -1,98 +0,0 @@ | |||||
| import unittest | |||||
| from d3m import container, utils | |||||
| from d3m.metadata import base as metadata_base | |||||
| from tods.timeseries_processing import SubsequenceClustering | |||||
| class SubsequenceClusteringTest(unittest.TestCase): | |||||
| def test_basic(self): | |||||
| self.maxDiff = None | |||||
| main = container.DataFrame({'a': [1., 2., 3.], 'b': [2., 3., 4.], 'c': [3., 4., 5.],}, | |||||
| # columns=['a', 'b', 'c'], | |||||
| generate_metadata=True) | |||||
| print(main) | |||||
| self.assertEqual(utils.to_json_structure(main.metadata.to_internal_simple_structure()), [{ | |||||
| 'selector': [], | |||||
| 'metadata': { | |||||
| # 'top_level': 'main', | |||||
| 'schema': metadata_base.CONTAINER_SCHEMA_VERSION, | |||||
| 'structural_type': 'd3m.container.pandas.DataFrame', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/Table'], | |||||
| 'dimension': { | |||||
| 'name': 'rows', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/TabularRow'], | |||||
| 'length': 3, | |||||
| }, | |||||
| }, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__'], | |||||
| 'metadata': { | |||||
| 'dimension': { | |||||
| 'name': 'columns', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/TabularColumn'], | |||||
| 'length': 3, | |||||
| }, | |||||
| }, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 0], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'a'}, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 1], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'b'}, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 2], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'c'} | |||||
| }]) | |||||
| self.assertIsInstance(main, container.DataFrame) | |||||
| hyperparams_class = SubsequenceClustering.SubsequenceClustering.metadata.get_hyperparams() | |||||
| primitive = SubsequenceClustering.SubsequenceClustering(hyperparams=hyperparams_class.defaults()) | |||||
| new_main = primitive.produce(inputs=main).value | |||||
| print(new_main) | |||||
| print(new_main.shape) | |||||
| self.assertEqual(utils.to_json_structure(main.metadata.to_internal_simple_structure()), [{ | |||||
| 'selector': [], | |||||
| 'metadata': { | |||||
| # 'top_level': 'main', | |||||
| 'schema': metadata_base.CONTAINER_SCHEMA_VERSION, | |||||
| 'structural_type': 'd3m.container.pandas.DataFrame', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/Table'], | |||||
| 'dimension': { | |||||
| 'name': 'rows', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/TabularRow'], | |||||
| 'length': 3, | |||||
| }, | |||||
| }, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__'], | |||||
| 'metadata': { | |||||
| 'dimension': { | |||||
| 'name': 'columns', | |||||
| 'semantic_types': ['https://metadata.datadrivendiscovery.org/types/TabularColumn'], | |||||
| 'length': 3, | |||||
| }, | |||||
| }, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 0], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'a'}, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 1], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'b'}, | |||||
| }, { | |||||
| 'selector': ['__ALL_ELEMENTS__', 2], | |||||
| 'metadata': {'structural_type': 'numpy.float64', 'name': 'c'} | |||||
| }]) | |||||
| if __name__ == '__main__': | |||||
| unittest.main() | |||||