def test_setup_epoch(self): column_info = {"type": types.DateTime()} bulkdata.setup_epoch(mock.Mock(), mock.Mock(), column_info) self.assertIsInstance(column_info["type"], bulkdata.EpochType)