Improve tests for insight.DataSet

Tests cover all the functions that are exposed.