https://github.com/puppetlabs/beaker/pull/108 is the `install_puppet` PR. I picture something like this in the `spec_helper_acceptance.rb`:
```ruby
  if host.is_pe?
    install_pe
  else
    install_puppet
  end
```

This should work, but I haven't gotten around to committing it to modules yet.