def test_load_table(self, mock_load_csv):
"""Test load table"""
# Setup
metadata = Mock(spec=Metadata)
metadata.root_path = 'a/path'
metadata.get_table_meta.return_value = {'some': 'data'}
mock_load_csv.return_value = 'data'
# Run
result = Metadata.load_table(metadata, 'test')
# Asserts
assert result == 'data'
metadata.get_table_meta.assert_called_once_with('test')
mock_load_csv.assert_called_once_with('a/path', {'some': 'data'})
def test_load_table(self, mock_load_csv):
"""Test load table"""
# Setup
root_path = '.'
table_meta = {'some': 'data'}
# Run
metadata = Mock()
metadata.root_path = root_path
metadata.get_table_meta.return_value = table_meta
mock_load_csv.return_value = 'data'
table_name = 'test'
result = Metadata.load_table(metadata, table_name)
# Asserts
metadata.get_table_meta.assert_called_once_with('test')
mock_load_csv.assert_called_once_with('.', {'some': 'data'})
assert result == 'data'