| author | Christian Urban <christian.urban@kcl.ac.uk> | 
| Fri, 05 Nov 2021 17:20:01 +0000 | |
| changeset 397 | 9755af1d74df | 
| 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  |