def test_get_dtypes_error_subtype_id(self):
"""Test get data types with an invalid id subtype."""
# Setup
table_meta = {'fields': {'item': {'type': 'id', 'subtype': 'boolean'}}}
metadata = Mock(spec=Metadata)
metadata.get_table_meta.return_value = table_meta
metadata._DTYPES = Metadata._DTYPES
# Run
with pytest.raises(ValueError):
Metadata.get_dtypes(metadata, 'test', ids=True)
def test_get_dtypes_error_id(self):
"""Test get data types with an id that is not a primary or foreign key."""
# Setup
table_meta = {'fields': {'item': {'type': 'id'}}}
metadata = Mock(spec=Metadata)
metadata.get_table_meta.return_value = table_meta
metadata._DTYPES = Metadata._DTYPES
# Run
with pytest.raises(ValueError):
Metadata.get_dtypes(metadata, 'test', ids=True)
def test_get_dtypes_error_invalid_type(self):
"""Test get data types with an invalid type."""
# Setup
table_meta = {'fields': {'item': {'type': 'unknown'}}}
metadata = Mock(spec=Metadata)
metadata.get_table_meta.return_value = table_meta
metadata._DTYPES = Metadata._DTYPES
# Run
with pytest.raises(ValueError):
Metadata.get_dtypes(metadata, 'test')
def test_get_dtypes_error_subtype_numerical(self):
"""Test get data types with an invalid numerical subtype."""
# Setup
table_meta = {
'fields': {
'item': {'type': 'numerical', 'subtype': 'boolean'}
}
}
metadata = Mock(spec_set=Metadata)
metadata.get_table_meta.return_value = table_meta
metadata._DTYPES = Metadata._DTYPES
# Run
with pytest.raises(MetadataError):
Metadata.get_dtypes(metadata, 'test')
def test_get_dtypes_no_ids(self):
"""Test get data types excluding ids."""
# Setup
table_meta = {
'fields': {
'item 0': {'type': 'id', 'subtype': 'integer'},
'item 1': {'type': 'numerical', 'subtype': 'integer'},
'item 2': {'type': 'numerical', 'subtype': 'float'},
'item 3': {'type': 'categorical'},
'item 4': {'type': 'boolean'},
'item 5': {'type': 'datetime'},
}
}
metadata = Mock(spec_set=Metadata)
metadata.get_table_meta.return_value = table_meta
metadata._DTYPES = Metadata._DTYPES
# Run
result = Metadata.get_dtypes(metadata, 'test')
# Asserts
expected = {
'item 1': 'int',
'item 2': 'float',
'item 3': 'object',
'item 4': 'bool',
'item 5': 'datetime64',
}
assert result == expected