| author | Christian Urban <christian.urban@kcl.ac.uk> | 
| Mon, 08 Nov 2021 01:03:09 +0000 | |
| changeset 401 | 317e49b9cab6 | 
| parent 379 | 73ad2e018516 | 
| child 403 | 312c9eb39ad8 | 
| permissions | -rw-r--r-- | 
| 211 | 1 | |
| 2 | import io.Source | |
| 3 | import scala.util._ | |
| 4 | ||
| 284 | 5 | import CW7b._ | 
| 6 | ||
| 211 | 7 | def urban_get_csv_file(name: String) : List[String] = {
 | 
| 379 | 8 |   val csv = Source.fromFile(name)("ISO-8859-1")
 | 
| 211 | 9 |   csv.mkString.split("\n").toList.drop(1)
 | 
| 10 | } | |
| 11 | ||
| 12 | val urban_ratings = urban_get_csv_file("ratings.csv")
 | |
| 13 | val urban_movies = urban_get_csv_file("movies.csv")
 | |
| 14 | ||
| 15 | ||
| 16 | assert(process_ratings(urban_ratings).length == 48580) | |
| 17 | ||
| 18 | assert(process_movies(urban_movies).length == 9742) | |
| 19 | ||
| 20 |