def test_expand_vector_cols(self):
"""Test the expand_vector_cols method."""
cols, dtypes, vals = expand_vector_cols(columns=COLS,
datatypes=DTYPES,
values=VALS)
self.assertEqual(cols, EXPANDED_COLS)
self.assertEqual(dtypes, EXPANDED_DTYPES)
self.assertEqual(vals, EXPANDED_VALS)
def _do_db_create(self, table_name, columns, datatypes, primary_key,
generate_pk, foreign_key, indexes):
"""Call db_create but expand the vectors first."""
columns, datatypes = expand_vector_cols(columns, datatypes)
check_characters(table_name, columns, datatypes, primary_key,
foreign_key, indexes)
self._db_create(table_name, columns, datatypes, primary_key,
generate_pk, foreign_key, indexes)
def _do_db_insert(self, table_name, columns, values, datatypes):
"""Call db_insert but expand vectors."""
columns, datatypes, values = expand_vector_cols(
columns, datatypes, values)
values = [
convert_from(v, datatypes.get(c)) for c, v in zip(columns, values)
]
check_characters(table_name, columns, datatypes)
return self._db_insert(table_name, columns, values, datatypes)
def _do_db_update(self, table_name, columns, values, condition, datatypes):
"""Call db_update but expand vectors."""
columns, datatypes, values = expand_vector_cols(
columns, datatypes, values)
condition = expand_vector_condition(condition)
values = [
convert_from(v, datatypes.get(c)) for c, v in zip(columns, values)
]
check_characters(table_name, columns, condition, datatypes)
self._db_update(table_name, columns, values, condition, datatypes)