diff options
| -rw-r--r-- | scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala b/scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala index 439b4f4..eb6f4ff 100644 --- a/scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala +++ b/scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala @@ -33,8 +33,8 @@ class HBaseMimeCountTest extends FunSpec with TupleConversions { List("sha1:095893C3YNNEGH5WAG5ZAAXWAEBNXJWT", mimeType2) ) - val mimeType1Count = 3 - val mimeType2Count = 5 + val mimeType1Count = sampleData.count(lst => lst(1) == mimeType1) + val mimeType2Count = sampleData.count(lst => lst(1) == mimeType2) JobTest("sandcrawler.HBaseMimeCountJob") .arg("test", "") |