self._inline_enum_script()
with capture_context_buffer() as buf:
command.upgrade(self.cfg, self.rid, sql=True)
- assert "CREATE TYPE pgenum AS ENUM ('one','two','three')" in buf.getvalue()
+ assert "CREATE TYPE pgenum AS ENUM ('one', 'two', 'three')" in buf.getvalue()
assert "CREATE TABLE sometable (\n data pgenum\n)" in buf.getvalue()
def test_offline_inline_enum_drop(self):
self._distinct_enum_script()
with capture_context_buffer() as buf:
command.upgrade(self.cfg, self.rid, sql=True)
- assert "CREATE TYPE pgenum AS ENUM ('one','two','three')" in buf.getvalue()
+ assert "CREATE TYPE pgenum AS ENUM ('one', 'two', 'three')" in buf.getvalue()
assert "CREATE TABLE sometable (\n data pgenum\n)" in buf.getvalue()
def test_offline_distinct_enum_drop(self):