From e808b3cad4a308039343d39126eb518fc404d987 Mon Sep 17 00:00:00 2001
From: redpony <redpony@ec762483-ff6d-05da-a07a-a48fb63a330f>
Date: Wed, 25 Aug 2010 16:24:44 +0000
Subject: onemore

git-svn-id: https://ws10smt.googlecode.com/svn/trunk@621 ec762483-ff6d-05da-a07a-a48fb63a330f
---
 gi/clda/src/ccrp.h      | 4 ++++
 gi/clda/src/crp_test.cc | 1 +
 2 files changed, 5 insertions(+)

(limited to 'gi/clda/src')

diff --git a/gi/clda/src/ccrp.h b/gi/clda/src/ccrp.h
index 9b1c5284..eeccce1a 100644
--- a/gi/clda/src/ccrp.h
+++ b/gi/clda/src/ccrp.h
@@ -30,6 +30,10 @@ class CCRP {
     return it->second.table_counts_.size();
   }
 
+  unsigned num_customers() const {
+    return num_customers_;
+  }
+
   // returns +1 or 0 indicating whether a new table was opened
   int increment(const Dish& dish, const double& p0, MT19937* rng) {
     DishLocations& loc = dish_locs_[dish];
diff --git a/gi/clda/src/crp_test.cc b/gi/clda/src/crp_test.cc
index 750d80a7..ed384f81 100644
--- a/gi/clda/src/crp_test.cc
+++ b/gi/clda/src/crp_test.cc
@@ -79,6 +79,7 @@ TEST_F(CRPTest, Exchangability) {
       ++hist[c];
       tot += c;
     }
+    EXPECT_EQ(cust, crp.num_customers());
     cerr << "P(a) = " << (xt / samples) << endl;
     cerr << "E[num tables] = " << (tot / samples) << endl;
     double error = fabs((tot / samples) - 5.4);
-- 
cgit v1.2.3