Speed up tests

* Use less epochs
* Use labels to limit source space
* Remove spurious test of using labels
* Modify expected values in asserts