def map_games(self):
if self.rdd is None or not isinstance(self.rdd, RDD):
raise Exception("Invalid RDD")
game_fields = MapGames.get_game_fields(self.rdd)
if game_fields is None or not isinstance(game_fields, RDD):
raise Exception("game_fields is None or not an instance of RDD")
mapped_games = MapGames.map_games(game_fields)
if mapped_games is None or not isinstance(mapped_games, RDD):
raise Exception("mapped_games is None or not an instance of RDD")
return mapped_games
def test_get_mapped_game_invalid_len(spark_context):
rdd = spark_context.parallelize(
["1,2,3,4,5,6", "10,20,30,40,50,60", "400,500,600,700,800,900"])
game_fields = MapGames.get_game_fields(rdd)
mapped_games = MapGames.map_games(game_fields).collect()
for r in mapped_games:
if r[0] == "1":
assert r[1] is None
if r[0] == "10":
assert r[1] is None
if r[0] == "100":
assert r[1] is None
def test_get_mapped_game(spark_context):
rdd = spark_context.parallelize(
["1,2,3,4,5,6,7,8,9", "10,20,30,40,50,60,70,80,90", "100,200,300,400,500,600,700,800,900"])
game_fields = MapGames.get_game_fields(rdd)
mapped_games = MapGames.map_games(game_fields).collect()
for r in mapped_games:
if r[0] == "1":
assert r[1].homeTeam == "8"
assert r[1].visitingTeam == "9"
assert r[1].season == "6"
if r[0] == "10":
assert r[1].homeTeam == "80"
assert r[1].visitingTeam == "90"
assert r[1].season == "60"
if r[0] == "100":
assert r[1].homeTeam == "800"
assert r[1].visitingTeam == "900"
assert r[1].season == "600"
def test_get_game_fields_null_rdd():
rdd = None
rdd_new = MapGames.get_game_fields(rdd)
assert rdd_new is None
def test_get_game_fields_empty_rdd(spark_context):
rdd = spark_context.emptyRDD()
rdd_new = MapGames.get_game_fields(rdd).collect()
assert len(rdd_new) == 0
def test_get_game_fields(spark_context):
rdd = spark_context.parallelize(
["1,2,3,4,5,6,7", "10,20,30,40,50,60,70", "100,200,300,400,500,600,700"])
rdd_new = MapGames.get_game_fields(rdd).collect()
assert(len(rdd_new) == 3)
def test_get_game_fields_not_rdd():
rdd = {}
rdd_new = MapGames.get_game_fields(rdd)
assert rdd_new is None