Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
#include "FWCore/Services/interface/ExternalRandomNumberGeneratorService.h"
#include "FWCore/Utilities/interface/StreamID.h"

#include "CLHEP/Random/JamesRandom.h"
#include "CLHEP/Random/RanecuEngine.h"
#include "CLHEP/Random/MixMaxRng.h"

//#define CATCH_CONFIG_MAIN
#include "catch.hpp"

namespace {
  void test(CLHEP::HepRandomEngine& iRand, CLHEP::HepRandomEngine& iEngine) {
    REQUIRE(iRand.flat() == iEngine.flat());
    REQUIRE(iRand.flat() == iEngine.flat());
    REQUIRE(iRand.flat() == iEngine.flat());
    REQUIRE(iRand.flat() == iEngine.flat());
  }
}  // namespace

TEST_CASE("Test ExternalRandomNumberGeneratorService", "[externalrandomnumbergeneratorservice]") {
  SECTION("JamesRandom") {
    edm::ExternalRandomNumberGeneratorService service;
    CLHEP::HepJamesRandom rand(12345);

    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));

    //advance the one to see how it works
    rand.flat();
    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));
  }

  SECTION("RanecuEngine") {
    edm::ExternalRandomNumberGeneratorService service;
    CLHEP::RanecuEngine rand(12345);

    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));

    //advance the one to see how it works
    rand.flat();
    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));
  }

  SECTION("MixMaxRng") {
    edm::ExternalRandomNumberGeneratorService service;
    CLHEP::MixMaxRng rand(12345);

    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));

    //advance the one to see how it works
    rand.flat();
    service.setState(rand.put(), rand.getSeed());
    test(rand, service.getEngine(edm::StreamID::invalidStreamID()));
  }
}