diff -r 663c2a9108d1 -r 4de31fdc0d67 main_testing2/danube_test5.scala --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/main_testing2/danube_test5.scala Mon Nov 02 02:31:44 2020 +0000 @@ -0,0 +1,29 @@ + +// first test + +import CW7b._ + +def urban_groupById(ratings: List[(String, String)]) = + ratings.groupBy(_._1).view.mapValues(_.map(_._2)).toMap + +def urban_get_csv_file(name: String) : List[String] = { + import io.Source + import scala.util._ + val csv = Source.fromFile(name) + csv.mkString.split("\n").toList.drop(1) +} + +def urban_process_ratings(lines: List[String]) : List[(String, String)] = { + for (cols <- lines.map(_.split(",").toList); + if (cols(2).toFloat >= 4)) yield (cols(0), cols(1)) +} + + +val urban_good_ratings = process_ratings(urban_get_csv_file("ratings.csv")) + +val urban_ratings_map = urban_groupById(urban_good_ratings) + +assert(suggestions(urban_ratings_map, "912").length == 4110) +assert(suggestions(urban_ratings_map, "858").length == 4883) +assert(suggestions(urban_ratings_map, "260").length == 4970) +