equal
deleted
inserted
replaced
1 |
1 |
2 import io.Source |
2 import io.Source |
3 import scala.util._ |
3 import scala.util._ |
4 |
4 |
5 import CW7b._ |
5 import M2._ |
6 |
6 |
7 def urban_get_csv_file(name: String) : List[String] = { |
7 def urban_get_csv_file(name: String) : List[String] = { |
8 val csv = Source.fromFile(name) |
8 val csv = Source.fromFile(name) |
9 csv.mkString.split("\n").toList.drop(1) |
9 csv.mkString.split("\n").toList.drop(1) |
10 } |
10 } |
11 |
11 |
12 val urban_ratings = urban_get_csv_file("ratings.csv") |
12 val urban_ratings = urban_get_csv_file("ratings.csv") |
13 val urban_movies = urban_get_csv_file("movies.csv") |
13 val urban_movies = urban_get_csv_file("movies.csv") |
14 |
14 |
|
15 val urban_res1 = process_ratings(urban_ratings) |
|
16 val urban_rat_elems = List(("1","1"), ("1","3"), ("1","6"), ("1","47")) |
15 |
17 |
16 assert(process_ratings(urban_ratings).length == 48580) |
18 assert(urban_res1.length == 48580) |
|
19 assert(urban_rat_elems.forall(urban_res1.contains(_))) |
17 |
20 |
18 assert(process_movies(urban_movies).length == 9742) |
21 assert(process_movies(urban_movies).length == 9742) |
19 |
22 |
20 |
23 |