Attic/danube_test4.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Mon, 25 Dec 2023 01:10:55 +0100
changeset 479 78cb5cdda3c3
parent 460 f5c0749858fd
permissions -rw-r--r--
updated jars
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
389
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     1
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     2
// first test 
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     3
421
Christian Urban <christian.urban@kcl.ac.uk>
parents: 389
diff changeset
     4
import M2._
389
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     5
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     6
def urban_groupById(ratings: List[(String, String)]) = 
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     7
  ratings.groupBy(_._1).view.mapValues(_.map(_._2)).toMap 
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     8
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
     9
def urban_get_csv_file(name: String) : List[String] = {
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    10
  import io.Source
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    11
  import scala.util._
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    12
  val csv = Source.fromFile(name)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    13
  csv.mkString.split("\n").toList.drop(1)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    14
}
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    15
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    16
def urban_process_ratings(lines: List[String]) : List[(String, String)] = {
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    17
  for (cols <- lines.map(_.split(",").toList); 
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    18
       if (cols(2).toFloat >= 4)) yield (cols(0), cols(1))  
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    19
}
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    20
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    21
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    22
val urban_good_ratings = process_ratings(urban_get_csv_file("ratings.csv"))
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    23
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    24
val urban_ratings_map = urban_groupById(urban_good_ratings)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    25
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    26
assert(favourites(urban_ratings_map, "912").length  == 80)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    27
assert(favourites(urban_ratings_map, "858").length  == 158)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    28
assert(favourites(urban_ratings_map, "260").length  == 201)
7e084cfbaac9 updated marking 2
Christian Urban <christian.urban@kcl.ac.uk>
parents:
diff changeset
    29