Change test_multi_column.py into a unittest.

Cherry-Picked from 84be61a3.