Browse Source

merge primitive_tests from Junjie branch, modified metadata for data_processing and timeseries_processing module

master
lhenry15 4 years ago
parent
commit
116a5dcb22
1 changed files with 0 additions and 98 deletions
  1. +0
    -98
      tods/tests/timeseries_processing/test_SubsequenceClustering.py

+ 0
- 98
tods/tests/timeseries_processing/test_SubsequenceClustering.py View File

@@ -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()

Loading…
Cancel
Save