From f81e193b4a2c7f61a46d1ead1bdf6b746997b3a8 Mon Sep 17 00:00:00 2001
From: Ellen Spertus <ellen.spertus@gmail.com>
Date: Wed, 6 Jun 2018 09:50:38 -0700
Subject: Made test data more robust.

---
 scalding/src/test/scala/sandcrawler/HBaseMimeCountTest.scala | 4 ++--
 1 file 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", "")
-- 
cgit v1.2.3