| author | Christian Urban <urbanc@in.tum.de> | 
| Wed, 07 Aug 2019 14:32:21 +0100 | |
| changeset 272 | a091de62e3f8 | 
| parent 211 | 1859d978b18e | 
| child 284 | fc20e5f83f0e | 
| permissions | -rw-r--r-- | 
| 211 | 1  | 
|
2  | 
import io.Source  | 
|
3  | 
import scala.util._  | 
|
4  | 
||
5  | 
def urban_get_csv_file(name: String) : List[String] = {
 | 
|
6  | 
val csv = Source.fromFile(name)  | 
|
7  | 
  csv.mkString.split("\n").toList.drop(1)
 | 
|
8  | 
}  | 
|
9  | 
||
10  | 
val urban_ratings = urban_get_csv_file("ratings.csv")
 | 
|
11  | 
val urban_movies = urban_get_csv_file("movies.csv")
 | 
|
12  | 
||
13  | 
||
14  | 
assert(process_ratings(urban_ratings).length == 48580)  | 
|
15  | 
||
16  | 
assert(process_movies(urban_movies).length == 9742)  | 
|
17  | 
||
18  |