mlab not working with mayavi in Ubuntu 16.04 (xenial)

I've installed mayavi using apt in Ubuntu 16.04 [sudo apt install mayavi2]. Unfortunately when trying to use, mlab seems to fail. I get the following error when running `ipython --gui=qt`:
`In[1]: from mayavi import mlab`
`(python:127406): Gtk-WARNING **: gtk_disable_setlocale() must be called before gtk_init()`

When running `ipython --gui=wx`:
`In[1]: from mayavi import mlab`
`In[2]: mlab.figure()`
`ipython:127461): Gdk-WARNING **: /build/gtk+2.0-KsZKkB/gtk+2.0-2.24.30/gdk/x11/gdkdrawable-x11.c:952 drawable is not a pixmap or window`
(this warning repeats 7 times before:)
`Out[2]: < mayavi.core.scene.Scene at 0x7f14001eaef0>`

I've run the nosetests for mayavi/tests and tvtk/tests and have attached the outputs along with the output for typing `mayavi2 -t` on the command line.
[mayavi2-t.txt](https://github.com/enthought/mayavi/files/558889/mayavi2-t.txt)
[nose_mayavi.txt](https://github.com/enthought/mayavi/files/558890/nose_mayavi.txt)
[nose_tvtk.txt](https://github.com/enthought/mayavi/files/558891/nose_tvtk.txt)

I've reproduced the same error(s) in Linux Mint 18 (Sarah), which is built on top of Ubuntu 16.04. The mayavi2 was just downloaded from the universe repository, so it's version 4.4.3-2.1.

Any help much appreciated.