testing2/danube_test2.scala
author Christian Urban <christian dot urban at kcl dot ac dot uk>
Tue, 06 Aug 2019 00:12:34 +0100
changeset 270 38e13601cb1b
parent 211 1859d978b18e
child 284 fc20e5f83f0e
permissions -rw-r--r--
updated


import io.Source
import scala.util._

def urban_get_csv_file(name: String) : List[String] = {
  val csv = Source.fromFile(name)
  csv.mkString.split("\n").toList.drop(1)
}

val urban_ratings = urban_get_csv_file("ratings.csv")
val urban_movies = urban_get_csv_file("movies.csv")


assert(process_ratings(urban_ratings).length == 48580)

assert(process_movies(urban_movies).length == 9742)