1 files changed, 6 insertions, 2 deletions
diff --git a/scalding/src/test/scala/sandcrawler/HBaseRowCountTest.scala b/scalding/src/test/scala/sandcrawler/HBaseRowCountTest.scala
index 598f45d..ac7cf18 100644
--- a/scalding/src/test/scala/sandcrawler/HBaseRowCountTest.scala
+++ b/scalding/src/test/scala/sandcrawler/HBaseRowCountTest.scala
@@ -46,11 +46,15 @@ class HBaseRowCountTest extends FunSpec with TupleConversions {
       sampleData.map(l => new Tuple(l.map(s => {new ImmutableBytesWritable(Bytes.toBytes(s))}):_*)))
     .sink[Tuple](Tsv(output format "get_list")) {
       outputBuffer =>
-        log.debug("Output => " + outputBuffer)
 
         it("should return the test data provided.") {
           println("outputBuffer.size => " + outputBuffer.size)
-          assert(outputBuffer.size === 2)
+          assert(outputBuffer.size === 1)
+        }
+
+        it("should return the correct count") {
+          println("raw output => " + outputBuffer)
+          assert(outputBuffer(0).getObject(0) === 2)
         }
     }
     .run