Add test case for simple partitioning

See #45.