Actually testing parted

called_once_with didn't work, changed to assert_called_once_with