avg as a double

Author: squito

src/main/scala/com/cloudera/sa/examples/tablestats/model/ColumnStats.scala

@@ -12,7 +12,7 @@ class ColumnStats(var nulls:Long = 0l,
                   var sumLong:Long = 0l,
                   val topNValues:TopNList = new TopNList(10)) extends Serializable {
 
-  def avgLong: Long = sumLong/totalCount
+  def avg: Double = sumLong / totalCount.toDouble
 
   //Part C.B
   def +=(colValue: Any, colCount: Long): Unit = {
@@ -49,6 +49,6 @@ class ColumnStats(var nulls:Long = 0l,
     }
   }
 
-  override def toString = s"ColumnStats(nulls=$nulls, empties=$empties, totalCount=$totalCount, uniqueValues=$uniqueValues, maxLong=$maxLong, minLong=$minLong, sumLong=$sumLong, topNValues=$topNValues, avgLong=$avgLong)"
+  override def toString = s"ColumnStats(nulls=$nulls, empties=$empties, totalCount=$totalCount, uniqueValues=$uniqueValues, maxLong=$maxLong, minLong=$minLong, sumLong=$sumLong, topNValues=$topNValues, avgLong=$avg)"
 }
 

src/test/scala/com/cloudera/sa/examples/tablestats/TestTableStatsSinglePathMain.scala

@@ -47,7 +47,7 @@ class TestTableStatsSinglePathMain extends FunSuite with BeforeAndAfterEach with
       assertResult(6l)(firstPassStats.columnStatsMap(0).maxLong)
       assertResult(1l)(firstPassStats.columnStatsMap(0).minLong)
       assertResult(21l)(firstPassStats.columnStatsMap(0).sumLong)
-      assertResult(3l)(firstPassStats.columnStatsMap(0).avgLong)
+      assertResult(3.5)(firstPassStats.columnStatsMap(0).avg)
 
       assertResult(2)(firstPassStats.columnStatsMap(3).topNValues.topNCountsForColumnArray.length)