Update test framework

Tests now pass on travis and locally (via the new Pkg.test()).