main_testing2/danube_test7.scala
changeset 384 6e1237691307
parent 379 5616b45d656f
--- a/main_testing2/danube_test7.scala	Mon Dec 07 01:25:41 2020 +0000
+++ b/main_testing2/danube_test7.scala	Fri Jan 15 02:40:57 2021 +0000
@@ -1,48 +1,13 @@
 
 import CW7b._
 
-// first test 
-
-def urban_groupById(ratings: List[(String, String)], 
-              m: Map[String, List[String]]) : Map[String, List[String]] = ratings match {
-  case Nil => m
-  case (id, mov) :: rest => {
-    val old_ratings = m.getOrElse (id, Nil)
-    val new_ratings = m + (id -> (mov :: old_ratings))
-    urban_groupById(rest, new_ratings)
-  }
-}
-//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)("ISO-8859-1")
-  csv.mkString.split("\n").toList.drop(1)
-}
+val urban_recs = 
+  Map("1" -> List("b", "a"), 
+      "2" -> List("y", "x"), 
+      "3" -> List("a", "c"))
 
-def urban_process_ratings(lines: List[String]) : List[(String, String)] = {
-  for (cols <- lines.map(_.split(",").toList); 
-       if (cols(2).toInt >= 4)) yield (cols(0), cols(1))  
-}
-
-def urban_process_movies(lines: List[String]) : List[(String, String)] = {
-  for (cols <- lines.map(_.split(",").toList)) yield (cols(0), cols(1))  
-}
-
+val urban_names = Map("a" -> "A", "b" -> "B", "c" -> "C", "x" -> "X", "y" -> "Y")
 
-val urban_good_ratings = urban_process_ratings(urban_get_csv_file("ratings.csv"))
-val urban_movie_names = urban_process_movies(urban_get_csv_file("movies.csv"))
-
-val urban_movie_names_map = urban_movie_names.toMap
-val urban_ratings_map = urban_groupById(urban_good_ratings, Map())
-//val urban_ratings_map = groupById(urban_good_ratings, Map())
+assert(most_recommended(urban_recs, urban_names).toSet ==
+        Set(("A",2), ("B",1), ("C",1), ("X",1), ("Y",1)))
 
-val urban_ratings_map2 =  for ((k, v) <- urban_ratings_map) yield (k, v.take(2))
-
-assert(most_recommended(urban_ratings_map2, urban_movie_names_map).take(3) == 
-       List(("M*A*S*H (a.k.a. MASH) (1970)",15), 
-            ("Star Trek: First Contact (1996)",10), 
-            ("Inception (2010)",9)))
-