From a52f65426e234a3a0cba5a7245696faf6aba6d6d Mon Sep 17 00:00:00 2001
From: Jeff Donahue <jeff.donahue@gmail.com>
Date: Mon, 7 Apr 2014 17:58:11 -0700
Subject: [PATCH] fix bernoulli*bernoulli test, now all pass

---
 src/caffe/test/test_random_number_generator.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/caffe/test/test_random_number_generator.cpp b/src/caffe/test/test_random_number_generator.cpp
index 25335c0..f33b4ce 100644
--- a/src/caffe/test/test_random_number_generator.cpp
+++ b/src/caffe/test/test_random_number_generator.cpp
@@ -326,6 +326,7 @@ TYPED_TEST(RandomNumberGeneratorTest, TestRngBernoulliTimesBernoulli) {
   SyncedMemory bernoulli1_data(sample_size * sizeof(int));
   SyncedMemory bernoulli2_data(sample_size * sizeof(int));
   Caffe::set_random_seed(1701);
+  // Sample from Bernoulli with p = 0.5
   TypeParam p1 = 0.5;
   caffe_vRngBernoulli(sample_size, static_cast<int*>(
       bernoulli1_data.mutable_cpu_data()), p1);
@@ -414,7 +415,7 @@ TYPED_TEST(RandomNumberGeneratorTest, TestRngBernoulliTimesBernoulli) {
   true_mean = p;
   true_std = sqrt(p * (1 - p));
   empirical_mean = this->sample_mean(
-      static_cast<const int *>(bernoulli2_data.cpu_data()), sample_size);
+      static_cast<const int *>(bernoulli1_data.cpu_data()), sample_size);
   bound = this->mean_bound(true_std, sample_size);
   LOG(INFO) << "Bernoulli1*Bernoulli2: Expected mean = " << true_mean
             << "; sample mean = " << empirical_mean;
-- 
2.7.4