Update test suite to prevent OOMs

Minimize data set to reduce memory pressure