diff --git a/tests/test_flatland_envs_city_generator.py b/tests/test_flatland_envs_city_generator.py
index 4a96a8fbf282fee79ccd778ba7d097042fbf31c3..fe39d785712c88f03af09c7a6d1dac715a585db3 100644
--- a/tests/test_flatland_envs_city_generator.py
+++ b/tests/test_flatland_envs_city_generator.py
@@ -26,8 +26,272 @@ def test_city_generator():
number_of_agents=10,
obs_builder_object=GlobalObsForRailEnv())
- # approximative test (replace image comparison)
- assert np.sum(env.rail.grid) == 3642337
+ expected_grid_map = np.zeros((50, 50), dtype=env.rail.transitions.get_type())
+
+ expected_grid_map[8][16]=4
+ expected_grid_map[8][17]=5633
+ expected_grid_map[8][18]=1025
+ expected_grid_map[8][19]=1025
+ expected_grid_map[8][20]=17411
+ expected_grid_map[8][21]=1025
+ expected_grid_map[8][22]=1025
+ expected_grid_map[8][23]=1025
+ expected_grid_map[8][24]=1025
+ expected_grid_map[8][25]=1025
+ expected_grid_map[8][26]=4608
+ expected_grid_map[9][16]=16386
+ expected_grid_map[9][17]=50211
+ expected_grid_map[9][18]=1025
+ expected_grid_map[9][19]=1025
+ expected_grid_map[9][20]=3089
+ expected_grid_map[9][21]=1025
+ expected_grid_map[9][22]=256
+ expected_grid_map[9][26]=32800
+ expected_grid_map[10][6]=16386
+ expected_grid_map[10][7]=1025
+ expected_grid_map[10][8]=1025
+ expected_grid_map[10][9]=1025
+ expected_grid_map[10][10]=1025
+ expected_grid_map[10][11]=1025
+ expected_grid_map[10][12]=1025
+ expected_grid_map[10][13]=1025
+ expected_grid_map[10][14]=1025
+ expected_grid_map[10][15]=1025
+ expected_grid_map[10][16]=33825
+ expected_grid_map[10][17]=34864
+ expected_grid_map[10][26]=32800
+ expected_grid_map[11][6]=32800
+ expected_grid_map[11][16]=32800
+ expected_grid_map[11][17]=32800
+ expected_grid_map[11][26]=32800
+ expected_grid_map[12][6]=32800
+ expected_grid_map[12][16]=32800
+ expected_grid_map[12][17]=32800
+ expected_grid_map[12][26]=32800
+ expected_grid_map[13][6]=32800
+ expected_grid_map[13][16]=32800
+ expected_grid_map[13][17]=32800
+ expected_grid_map[13][26]=32800
+ expected_grid_map[14][6]=32800
+ expected_grid_map[14][16]=32800
+ expected_grid_map[14][17]=32800
+ expected_grid_map[14][26]=32800
+ expected_grid_map[15][6]=32800
+ expected_grid_map[15][16]=32800
+ expected_grid_map[15][17]=32800
+ expected_grid_map[15][26]=32800
+ expected_grid_map[16][6]=32800
+ expected_grid_map[16][16]=32800
+ expected_grid_map[16][17]=32800
+ expected_grid_map[16][26]=32800
+ expected_grid_map[17][6]=32800
+ expected_grid_map[17][16]=72
+ expected_grid_map[17][17]=1097
+ expected_grid_map[17][18]=1025
+ expected_grid_map[17][19]=1025
+ expected_grid_map[17][20]=1025
+ expected_grid_map[17][21]=1025
+ expected_grid_map[17][22]=1025
+ expected_grid_map[17][23]=1025
+ expected_grid_map[17][24]=1025
+ expected_grid_map[17][25]=1025
+ expected_grid_map[17][26]=33825
+ expected_grid_map[17][27]=4608
+ expected_grid_map[18][6]=32800
+ expected_grid_map[18][26]=72
+ expected_grid_map[18][27]=52275
+ expected_grid_map[18][28]=5633
+ expected_grid_map[18][29]=17411
+ expected_grid_map[18][30]=1025
+ expected_grid_map[18][31]=1025
+ expected_grid_map[18][32]=256
+ expected_grid_map[19][6]=32800
+ expected_grid_map[19][25]=16386
+ expected_grid_map[19][26]=1025
+ expected_grid_map[19][27]=2136
+ expected_grid_map[19][28]=1097
+ expected_grid_map[19][29]=1097
+ expected_grid_map[19][30]=5633
+ expected_grid_map[19][31]=1025
+ expected_grid_map[19][32]=256
+ expected_grid_map[20][6]=32800
+ expected_grid_map[20][25]=32800
+ expected_grid_map[20][26]=16386
+ expected_grid_map[20][27]=17411
+ expected_grid_map[20][28]=1025
+ expected_grid_map[20][29]=1025
+ expected_grid_map[20][30]=3089
+ expected_grid_map[20][31]=1025
+ expected_grid_map[20][32]=256
+ expected_grid_map[21][6]=32800
+ expected_grid_map[21][16]=16386
+ expected_grid_map[21][17]=1025
+ expected_grid_map[21][18]=1025
+ expected_grid_map[21][19]=1025
+ expected_grid_map[21][20]=1025
+ expected_grid_map[21][21]=1025
+ expected_grid_map[21][22]=1025
+ expected_grid_map[21][23]=1025
+ expected_grid_map[21][24]=1025
+ expected_grid_map[21][25]=33825
+ expected_grid_map[21][26]=33825
+ expected_grid_map[21][27]=2064
+ expected_grid_map[22][6]=32800
+ expected_grid_map[22][16]=32800
+ expected_grid_map[22][25]=32800
+ expected_grid_map[22][26]=32800
+ expected_grid_map[23][6]=32800
+ expected_grid_map[23][16]=32800
+ expected_grid_map[23][25]=32800
+ expected_grid_map[23][26]=32800
+ expected_grid_map[24][6]=32800
+ expected_grid_map[24][16]=32800
+ expected_grid_map[24][25]=32800
+ expected_grid_map[24][26]=32800
+ expected_grid_map[25][6]=32800
+ expected_grid_map[25][16]=32800
+ expected_grid_map[25][25]=32800
+ expected_grid_map[25][26]=32800
+ expected_grid_map[26][6]=32800
+ expected_grid_map[26][16]=32800
+ expected_grid_map[26][25]=32800
+ expected_grid_map[26][26]=32800
+ expected_grid_map[27][6]=72
+ expected_grid_map[27][7]=1025
+ expected_grid_map[27][8]=1025
+ expected_grid_map[27][9]=17411
+ expected_grid_map[27][10]=1025
+ expected_grid_map[27][11]=1025
+ expected_grid_map[27][12]=1025
+ expected_grid_map[27][13]=1025
+ expected_grid_map[27][14]=1025
+ expected_grid_map[27][15]=4608
+ expected_grid_map[27][16]=72
+ expected_grid_map[27][17]=17411
+ expected_grid_map[27][18]=5633
+ expected_grid_map[27][19]=1025
+ expected_grid_map[27][20]=1025
+ expected_grid_map[27][21]=1025
+ expected_grid_map[27][22]=1025
+ expected_grid_map[27][23]=1025
+ expected_grid_map[27][24]=1025
+ expected_grid_map[27][25]=33825
+ expected_grid_map[27][26]=2064
+ expected_grid_map[28][6]=4
+ expected_grid_map[28][7]=1025
+ expected_grid_map[28][8]=1025
+ expected_grid_map[28][9]=3089
+ expected_grid_map[28][10]=1025
+ expected_grid_map[28][11]=1025
+ expected_grid_map[28][12]=1025
+ expected_grid_map[28][13]=1025
+ expected_grid_map[28][14]=4608
+ expected_grid_map[28][15]=72
+ expected_grid_map[28][16]=1025
+ expected_grid_map[28][17]=2136
+ expected_grid_map[28][18]=1097
+ expected_grid_map[28][19]=5633
+ expected_grid_map[28][20]=5633
+ expected_grid_map[28][21]=1025
+ expected_grid_map[28][22]=256
+ expected_grid_map[28][25]=32800
+ expected_grid_map[29][6]=4
+ expected_grid_map[29][7]=5633
+ expected_grid_map[29][8]=20994
+ expected_grid_map[29][9]=5633
+ expected_grid_map[29][10]=1025
+ expected_grid_map[29][11]=1025
+ expected_grid_map[29][12]=1025
+ expected_grid_map[29][13]=1025
+ expected_grid_map[29][14]=1097
+ expected_grid_map[29][15]=5633
+ expected_grid_map[29][16]=1025
+ expected_grid_map[29][17]=17411
+ expected_grid_map[29][18]=5633
+ expected_grid_map[29][19]=1097
+ expected_grid_map[29][20]=3089
+ expected_grid_map[29][21]=20994
+ expected_grid_map[29][22]=1025
+ expected_grid_map[29][23]=1025
+ expected_grid_map[29][24]=1025
+ expected_grid_map[29][25]=2064
+ expected_grid_map[30][6]=16386
+ expected_grid_map[30][7]=38505
+ expected_grid_map[30][8]=3089
+ expected_grid_map[30][9]=1097
+ expected_grid_map[30][10]=1025
+ expected_grid_map[30][11]=1025
+ expected_grid_map[30][12]=256
+ expected_grid_map[30][15]=32800
+ expected_grid_map[30][16]=16386
+ expected_grid_map[30][17]=52275
+ expected_grid_map[30][18]=1097
+ expected_grid_map[30][19]=1025
+ expected_grid_map[30][20]=1025
+ expected_grid_map[30][21]=3089
+ expected_grid_map[30][22]=256
+ expected_grid_map[31][6]=32800
+ expected_grid_map[31][7]=32800
+ expected_grid_map[31][15]=72
+ expected_grid_map[31][16]=37408
+ expected_grid_map[31][17]=32800
+ expected_grid_map[32][6]=32800
+ expected_grid_map[32][7]=32800
+ expected_grid_map[32][16]=32800
+ expected_grid_map[32][17]=32800
+ expected_grid_map[33][6]=32800
+ expected_grid_map[33][7]=32800
+ expected_grid_map[33][16]=32800
+ expected_grid_map[33][17]=32800
+ expected_grid_map[34][6]=32800
+ expected_grid_map[34][7]=32800
+ expected_grid_map[34][16]=32800
+ expected_grid_map[34][17]=32800
+ expected_grid_map[35][6]=32800
+ expected_grid_map[35][7]=32800
+ expected_grid_map[35][16]=32800
+ expected_grid_map[35][17]=32800
+ expected_grid_map[36][6]=32800
+ expected_grid_map[36][7]=32800
+ expected_grid_map[36][16]=32800
+ expected_grid_map[36][17]=32800
+ expected_grid_map[37][6]=72
+ expected_grid_map[37][7]=1097
+ expected_grid_map[37][8]=1025
+ expected_grid_map[37][9]=1025
+ expected_grid_map[37][10]=1025
+ expected_grid_map[37][11]=1025
+ expected_grid_map[37][12]=1025
+ expected_grid_map[37][13]=1025
+ expected_grid_map[37][14]=1025
+ expected_grid_map[37][15]=1025
+ expected_grid_map[37][16]=33897
+ expected_grid_map[37][17]=37408
+ expected_grid_map[38][16]=72
+ expected_grid_map[38][17]=52275
+ expected_grid_map[38][18]=5633
+ expected_grid_map[38][19]=17411
+ expected_grid_map[38][20]=1025
+ expected_grid_map[38][21]=1025
+ expected_grid_map[38][22]=256
+ expected_grid_map[39][16]=4
+ expected_grid_map[39][17]=52275
+ expected_grid_map[39][18]=3089
+ expected_grid_map[39][19]=1097
+ expected_grid_map[39][20]=5633
+ expected_grid_map[39][21]=1025
+ expected_grid_map[39][22]=256
+ expected_grid_map[40][16]=4
+ expected_grid_map[40][17]=1097
+ expected_grid_map[40][18]=1025
+ expected_grid_map[40][19]=1025
+ expected_grid_map[40][20]=3089
+ expected_grid_map[40][21]=1025
+ expected_grid_map[40][22]=256
+
+ assert np.array_equal(env.rail.grid, expected_grid_map), "actual={}, expected={}".format(env.rail.grid,
+ expected_grid_map)
+
s0 = 0
s1 = 0
for a in range(env.get_num_agents()):
diff --git a/tests/test_flatland_envs_sparse_rail_generator.py b/tests/test_flatland_envs_sparse_rail_generator.py
index 47cb174f6938fddccae8b3374cbaeb73f31e2c67..e164752483e2b4ad5896d754d378a5519c960237 100644
--- a/tests/test_flatland_envs_sparse_rail_generator.py
+++ b/tests/test_flatland_envs_sparse_rail_generator.py
@@ -25,10 +25,687 @@ def test_sparse_rail_generator():
schedule_generator=sparse_schedule_generator(),
number_of_agents=10,
obs_builder_object=GlobalObsForRailEnv())
+ expected_grid_map = np.zeros((50, 50), dtype=env.rail.transitions.get_type())
+ expected_grid_map[1][33] = 8192
+ expected_grid_map[2][33] = 32800
+ expected_grid_map[3][31] = 4
+ expected_grid_map[3][32] = 4608
+ expected_grid_map[3][33] = 32800
+ expected_grid_map[4][30] = 16386
+ expected_grid_map[4][31] = 17411
+ expected_grid_map[4][32] = 1097
+ expected_grid_map[4][33] = 38505
+ expected_grid_map[4][34] = 1025
+ expected_grid_map[4][35] = 5633
+ expected_grid_map[4][36] = 1025
+ expected_grid_map[4][37] = 1025
+ expected_grid_map[4][38] = 1025
+ expected_grid_map[4][39] = 1025
+ expected_grid_map[4][40] = 1025
+ expected_grid_map[4][41] = 1025
+ expected_grid_map[4][42] = 1025
+ expected_grid_map[4][43] = 1025
+ expected_grid_map[4][44] = 1025
+ expected_grid_map[4][45] = 1025
+ expected_grid_map[4][46] = 1025
+ expected_grid_map[4][47] = 1025
+ expected_grid_map[4][48] = 4608
+ expected_grid_map[5][30] = 128
+ expected_grid_map[5][31] = 32800
+ expected_grid_map[5][33] = 32800
+ expected_grid_map[5][35] = 32800
+ expected_grid_map[5][48] = 32800
+ expected_grid_map[6][30] = 4
+ expected_grid_map[6][31] = 2064
+ expected_grid_map[6][33] = 32800
+ expected_grid_map[6][35] = 128
+ expected_grid_map[6][48] = 32800
+ expected_grid_map[7][33] = 32872
+ expected_grid_map[7][34] = 1025
+ expected_grid_map[7][35] = 1025
+ expected_grid_map[7][36] = 1025
+ expected_grid_map[7][37] = 1025
+ expected_grid_map[7][38] = 1025
+ expected_grid_map[7][39] = 1025
+ expected_grid_map[7][40] = 1025
+ expected_grid_map[7][41] = 20994
+ expected_grid_map[7][42] = 1025
+ expected_grid_map[7][43] = 1025
+ expected_grid_map[7][44] = 1025
+ expected_grid_map[7][45] = 1025
+ expected_grid_map[7][46] = 1025
+ expected_grid_map[7][47] = 4608
+ expected_grid_map[7][48] = 32800
+ expected_grid_map[8][3] = 16386
+ expected_grid_map[8][4] = 1025
+ expected_grid_map[8][5] = 1025
+ expected_grid_map[8][6] = 1025
+ expected_grid_map[8][7] = 1025
+ expected_grid_map[8][8] = 1025
+ expected_grid_map[8][9] = 1025
+ expected_grid_map[8][10] = 1025
+ expected_grid_map[8][11] = 5633
+ expected_grid_map[8][12] = 1025
+ expected_grid_map[8][13] = 1025
+ expected_grid_map[8][14] = 1025
+ expected_grid_map[8][15] = 1025
+ expected_grid_map[8][16] = 1025
+ expected_grid_map[8][17] = 1025
+ expected_grid_map[8][18] = 4608
+ expected_grid_map[8][33] = 32800
+ expected_grid_map[8][41] = 32800
+ expected_grid_map[8][47] = 32800
+ expected_grid_map[8][48] = 32800
+ expected_grid_map[9][3] = 32800
+ expected_grid_map[9][11] = 32800
+ expected_grid_map[9][12] = 8192
+ expected_grid_map[9][13] = 8192
+ expected_grid_map[9][18] = 32800
+ expected_grid_map[9][33] = 32800
+ expected_grid_map[9][41] = 32800
+ expected_grid_map[9][47] = 32800
+ expected_grid_map[9][48] = 32800
+ expected_grid_map[10][3] = 32800
+ expected_grid_map[10][8] = 8192
+ expected_grid_map[10][11] = 32800
+ expected_grid_map[10][12] = 32800
+ expected_grid_map[10][13] = 32800
+ expected_grid_map[10][18] = 32800
+ expected_grid_map[10][33] = 32800
+ expected_grid_map[10][41] = 32800
+ expected_grid_map[10][47] = 32800
+ expected_grid_map[10][48] = 32800
+ expected_grid_map[11][3] = 32800
+ expected_grid_map[11][8] = 32800
+ expected_grid_map[11][11] = 32800
+ expected_grid_map[11][12] = 32800
+ expected_grid_map[11][13] = 32800
+ expected_grid_map[11][18] = 32800
+ expected_grid_map[11][33] = 32800
+ expected_grid_map[11][41] = 32800
+ expected_grid_map[11][47] = 32800
+ expected_grid_map[11][48] = 32800
+ expected_grid_map[12][3] = 32800
+ expected_grid_map[12][8] = 72
+ expected_grid_map[12][9] = 1025
+ expected_grid_map[12][10] = 17411
+ expected_grid_map[12][11] = 52275
+ expected_grid_map[12][12] = 3089
+ expected_grid_map[12][13] = 3089
+ expected_grid_map[12][14] = 1025
+ expected_grid_map[12][15] = 1025
+ expected_grid_map[12][16] = 1025
+ expected_grid_map[12][17] = 1025
+ expected_grid_map[12][18] = 33825
+ expected_grid_map[12][19] = 1025
+ expected_grid_map[12][20] = 1025
+ expected_grid_map[12][21] = 1025
+ expected_grid_map[12][22] = 1025
+ expected_grid_map[12][23] = 1025
+ expected_grid_map[12][24] = 1025
+ expected_grid_map[12][25] = 1025
+ expected_grid_map[12][26] = 1025
+ expected_grid_map[12][27] = 1025
+ expected_grid_map[12][28] = 1025
+ expected_grid_map[12][29] = 1025
+ expected_grid_map[12][30] = 1025
+ expected_grid_map[12][31] = 1025
+ expected_grid_map[12][32] = 1025
+ expected_grid_map[12][33] = 33825
+ expected_grid_map[12][34] = 1025
+ expected_grid_map[12][35] = 1025
+ expected_grid_map[12][36] = 1025
+ expected_grid_map[12][37] = 1025
+ expected_grid_map[12][38] = 1025
+ expected_grid_map[12][39] = 1025
+ expected_grid_map[12][40] = 1025
+ expected_grid_map[12][41] = 35889
+ expected_grid_map[12][42] = 4608
+ expected_grid_map[12][47] = 32800
+ expected_grid_map[12][48] = 32800
+ expected_grid_map[13][3] = 32800
+ expected_grid_map[13][10] = 32800
+ expected_grid_map[13][11] = 32872
+ expected_grid_map[13][12] = 1025
+ expected_grid_map[13][13] = 256
+ expected_grid_map[13][15] = 8192
+ expected_grid_map[13][16] = 8192
+ expected_grid_map[13][17] = 8192
+ expected_grid_map[13][18] = 32800
+ expected_grid_map[13][20] = 8192
+ expected_grid_map[13][33] = 32800
+ expected_grid_map[13][41] = 32800
+ expected_grid_map[13][42] = 32800
+ expected_grid_map[13][47] = 32800
+ expected_grid_map[13][48] = 32800
+ expected_grid_map[14][3] = 32800
+ expected_grid_map[14][10] = 128
+ expected_grid_map[14][11] = 32800
+ expected_grid_map[14][15] = 72
+ expected_grid_map[14][16] = 37408
+ expected_grid_map[14][17] = 32800
+ expected_grid_map[14][18] = 32800
+ expected_grid_map[14][20] = 32800
+ expected_grid_map[14][33] = 32800
+ expected_grid_map[14][41] = 32800
+ expected_grid_map[14][42] = 32800
+ expected_grid_map[14][47] = 32800
+ expected_grid_map[14][48] = 32800
+ expected_grid_map[15][3] = 32800
+ expected_grid_map[15][11] = 32800
+ expected_grid_map[15][15] = 4
+ expected_grid_map[15][16] = 1097
+ expected_grid_map[15][17] = 1097
+ expected_grid_map[15][18] = 3089
+ expected_grid_map[15][19] = 1025
+ expected_grid_map[15][20] = 3089
+ expected_grid_map[15][21] = 1025
+ expected_grid_map[15][22] = 1025
+ expected_grid_map[15][23] = 1025
+ expected_grid_map[15][24] = 1025
+ expected_grid_map[15][25] = 1025
+ expected_grid_map[15][26] = 1025
+ expected_grid_map[15][27] = 1025
+ expected_grid_map[15][28] = 1025
+ expected_grid_map[15][29] = 1025
+ expected_grid_map[15][30] = 1025
+ expected_grid_map[15][31] = 1025
+ expected_grid_map[15][32] = 1025
+ expected_grid_map[15][33] = 33825
+ expected_grid_map[15][34] = 1025
+ expected_grid_map[15][35] = 1025
+ expected_grid_map[15][36] = 1025
+ expected_grid_map[15][37] = 1025
+ expected_grid_map[15][38] = 1025
+ expected_grid_map[15][39] = 1025
+ expected_grid_map[15][40] = 1025
+ expected_grid_map[15][41] = 35889
+ expected_grid_map[15][42] = 37408
+ expected_grid_map[15][47] = 32800
+ expected_grid_map[15][48] = 32800
+ expected_grid_map[16][3] = 32800
+ expected_grid_map[16][7] = 8192
+ expected_grid_map[16][11] = 32800
+ expected_grid_map[16][33] = 32800
+ expected_grid_map[16][41] = 32800
+ expected_grid_map[16][42] = 32800
+ expected_grid_map[16][47] = 32800
+ expected_grid_map[16][48] = 32800
+ expected_grid_map[17][3] = 32800
+ expected_grid_map[17][7] = 32800
+ expected_grid_map[17][9] = 8192
+ expected_grid_map[17][10] = 8192
+ expected_grid_map[17][11] = 32800
+ expected_grid_map[17][33] = 32800
+ expected_grid_map[17][41] = 32800
+ expected_grid_map[17][42] = 32800
+ expected_grid_map[17][47] = 32800
+ expected_grid_map[17][48] = 32800
+ expected_grid_map[18][3] = 32800
+ expected_grid_map[18][7] = 32800
+ expected_grid_map[18][8] = 8192
+ expected_grid_map[18][9] = 32800
+ expected_grid_map[18][10] = 32800
+ expected_grid_map[18][11] = 32800
+ expected_grid_map[18][33] = 32800
+ expected_grid_map[18][41] = 32800
+ expected_grid_map[18][42] = 32800
+ expected_grid_map[18][47] = 32800
+ expected_grid_map[18][48] = 32800
+ expected_grid_map[19][3] = 72
+ expected_grid_map[19][4] = 1025
+ expected_grid_map[19][5] = 1025
+ expected_grid_map[19][6] = 1025
+ expected_grid_map[19][7] = 1097
+ expected_grid_map[19][8] = 1097
+ expected_grid_map[19][9] = 1097
+ expected_grid_map[19][10] = 52275
+ expected_grid_map[19][11] = 33825
+ expected_grid_map[19][12] = 1025
+ expected_grid_map[19][13] = 1025
+ expected_grid_map[19][14] = 4608
+ expected_grid_map[19][33] = 32800
+ expected_grid_map[19][41] = 32800
+ expected_grid_map[19][42] = 32800
+ expected_grid_map[19][47] = 32800
+ expected_grid_map[19][48] = 32800
+ expected_grid_map[20][7] = 4
+ expected_grid_map[20][8] = 1025
+ expected_grid_map[20][9] = 1025
+ expected_grid_map[20][10] = 34864
+ expected_grid_map[20][11] = 32800
+ expected_grid_map[20][14] = 32800
+ expected_grid_map[20][33] = 32800
+ expected_grid_map[20][41] = 32800
+ expected_grid_map[20][42] = 32800
+ expected_grid_map[20][47] = 32800
+ expected_grid_map[20][48] = 32800
+ expected_grid_map[21][10] = 32800
+ expected_grid_map[21][11] = 32800
+ expected_grid_map[21][14] = 32800
+ expected_grid_map[21][24] = 8192
+ expected_grid_map[21][33] = 32872
+ expected_grid_map[21][34] = 1025
+ expected_grid_map[21][35] = 1025
+ expected_grid_map[21][36] = 1025
+ expected_grid_map[21][37] = 1025
+ expected_grid_map[21][38] = 1025
+ expected_grid_map[21][39] = 1025
+ expected_grid_map[21][40] = 1025
+ expected_grid_map[21][41] = 33825
+ expected_grid_map[21][42] = 38505
+ expected_grid_map[21][43] = 1025
+ expected_grid_map[21][44] = 1025
+ expected_grid_map[21][45] = 1025
+ expected_grid_map[21][46] = 1025
+ expected_grid_map[21][47] = 37408
+ expected_grid_map[21][48] = 32800
+ expected_grid_map[22][10] = 32800
+ expected_grid_map[22][11] = 32800
+ expected_grid_map[22][14] = 32800
+ expected_grid_map[22][22] = 8192
+ expected_grid_map[22][24] = 32800
+ expected_grid_map[22][27] = 8192
+ expected_grid_map[22][33] = 32800
+ expected_grid_map[22][41] = 32800
+ expected_grid_map[22][42] = 32800
+ expected_grid_map[22][47] = 32800
+ expected_grid_map[22][48] = 32800
+ expected_grid_map[23][10] = 32800
+ expected_grid_map[23][11] = 32800
+ expected_grid_map[23][14] = 32800
+ expected_grid_map[23][22] = 72
+ expected_grid_map[23][23] = 17411
+ expected_grid_map[23][24] = 1097
+ expected_grid_map[23][25] = 17411
+ expected_grid_map[23][26] = 1025
+ expected_grid_map[23][27] = 3089
+ expected_grid_map[23][28] = 1025
+ expected_grid_map[23][29] = 1025
+ expected_grid_map[23][30] = 1025
+ expected_grid_map[23][31] = 1025
+ expected_grid_map[23][32] = 1025
+ expected_grid_map[23][33] = 33825
+ expected_grid_map[23][34] = 1025
+ expected_grid_map[23][35] = 1025
+ expected_grid_map[23][36] = 1025
+ expected_grid_map[23][37] = 1025
+ expected_grid_map[23][38] = 1025
+ expected_grid_map[23][39] = 1025
+ expected_grid_map[23][40] = 1025
+ expected_grid_map[23][41] = 3089
+ expected_grid_map[23][42] = 34864
+ expected_grid_map[23][47] = 32800
+ expected_grid_map[23][48] = 32800
+ expected_grid_map[24][10] = 32800
+ expected_grid_map[24][11] = 32800
+ expected_grid_map[24][14] = 32800
+ expected_grid_map[24][23] = 32800
+ expected_grid_map[24][24] = 4
+ expected_grid_map[24][25] = 34864
+ expected_grid_map[24][33] = 32800
+ expected_grid_map[24][42] = 32800
+ expected_grid_map[24][47] = 32800
+ expected_grid_map[24][48] = 32800
+ expected_grid_map[25][10] = 32800
+ expected_grid_map[25][11] = 32800
+ expected_grid_map[25][14] = 32800
+ expected_grid_map[25][23] = 128
+ expected_grid_map[25][25] = 32800
+ expected_grid_map[25][33] = 32800
+ expected_grid_map[25][42] = 32800
+ expected_grid_map[25][47] = 32800
+ expected_grid_map[25][48] = 32800
+ expected_grid_map[26][10] = 32800
+ expected_grid_map[26][11] = 32800
+ expected_grid_map[26][14] = 32800
+ expected_grid_map[26][25] = 32800
+ expected_grid_map[26][33] = 32800
+ expected_grid_map[26][42] = 32800
+ expected_grid_map[26][47] = 32800
+ expected_grid_map[26][48] = 32800
+ expected_grid_map[27][10] = 32800
+ expected_grid_map[27][11] = 32800
+ expected_grid_map[27][14] = 32800
+ expected_grid_map[27][25] = 32800
+ expected_grid_map[27][33] = 32800
+ expected_grid_map[27][42] = 32800
+ expected_grid_map[27][47] = 32800
+ expected_grid_map[27][48] = 32800
+ expected_grid_map[28][10] = 32800
+ expected_grid_map[28][11] = 32800
+ expected_grid_map[28][14] = 32800
+ expected_grid_map[28][25] = 32800
+ expected_grid_map[28][33] = 49186
+ expected_grid_map[28][34] = 256
+ expected_grid_map[28][42] = 32800
+ expected_grid_map[28][44] = 8192
+ expected_grid_map[28][45] = 8192
+ expected_grid_map[28][47] = 32800
+ expected_grid_map[28][48] = 32800
+ expected_grid_map[28][49] = 8192
+ expected_grid_map[29][10] = 32800
+ expected_grid_map[29][11] = 32800
+ expected_grid_map[29][14] = 32800
+ expected_grid_map[29][25] = 32800
+ expected_grid_map[29][32] = 16386
+ expected_grid_map[29][33] = 37408
+ expected_grid_map[29][34] = 8192
+ expected_grid_map[29][42] = 32800
+ expected_grid_map[29][44] = 72
+ expected_grid_map[29][45] = 37408
+ expected_grid_map[29][47] = 32800
+ expected_grid_map[29][48] = 32800
+ expected_grid_map[29][49] = 32800
+ expected_grid_map[30][10] = 32800
+ expected_grid_map[30][11] = 32800
+ expected_grid_map[30][14] = 32800
+ expected_grid_map[30][25] = 32800
+ expected_grid_map[30][32] = 128
+ expected_grid_map[30][33] = 49186
+ expected_grid_map[30][34] = 33825
+ expected_grid_map[30][35] = 1025
+ expected_grid_map[30][36] = 1025
+ expected_grid_map[30][37] = 1025
+ expected_grid_map[30][38] = 1025
+ expected_grid_map[30][39] = 5633
+ expected_grid_map[30][40] = 1025
+ expected_grid_map[30][41] = 1025
+ expected_grid_map[30][42] = 2064
+ expected_grid_map[30][45] = 16458
+ expected_grid_map[30][46] = 17411
+ expected_grid_map[30][47] = 38505
+ expected_grid_map[30][48] = 38433
+ expected_grid_map[30][49] = 2064
+ expected_grid_map[31][10] = 32800
+ expected_grid_map[31][11] = 32800
+ expected_grid_map[31][14] = 32800
+ expected_grid_map[31][25] = 32800
+ expected_grid_map[31][30] = 8192
+ expected_grid_map[31][31] = 4
+ expected_grid_map[31][32] = 17411
+ expected_grid_map[31][33] = 34864
+ expected_grid_map[31][34] = 32800
+ expected_grid_map[31][39] = 32800
+ expected_grid_map[31][45] = 32800
+ expected_grid_map[31][46] = 32800
+ expected_grid_map[31][47] = 32800
+ expected_grid_map[31][48] = 32800
+ expected_grid_map[32][10] = 32800
+ expected_grid_map[32][11] = 32800
+ expected_grid_map[32][14] = 32800
+ expected_grid_map[32][25] = 32800
+ expected_grid_map[32][30] = 72
+ expected_grid_map[32][31] = 1025
+ expected_grid_map[32][32] = 2064
+ expected_grid_map[32][33] = 32872
+ expected_grid_map[32][34] = 2064
+ expected_grid_map[32][39] = 32800
+ expected_grid_map[32][45] = 128
+ expected_grid_map[32][46] = 128
+ expected_grid_map[32][47] = 32800
+ expected_grid_map[32][48] = 32800
+ expected_grid_map[33][10] = 32800
+ expected_grid_map[33][11] = 32800
+ expected_grid_map[33][14] = 32872
+ expected_grid_map[33][15] = 1025
+ expected_grid_map[33][16] = 1025
+ expected_grid_map[33][17] = 1025
+ expected_grid_map[33][18] = 1025
+ expected_grid_map[33][19] = 1025
+ expected_grid_map[33][20] = 1025
+ expected_grid_map[33][21] = 1025
+ expected_grid_map[33][22] = 1025
+ expected_grid_map[33][23] = 1025
+ expected_grid_map[33][24] = 1025
+ expected_grid_map[33][25] = 35889
+ expected_grid_map[33][26] = 1025
+ expected_grid_map[33][27] = 1025
+ expected_grid_map[33][28] = 1025
+ expected_grid_map[33][29] = 1025
+ expected_grid_map[33][30] = 1025
+ expected_grid_map[33][31] = 1025
+ expected_grid_map[33][32] = 1025
+ expected_grid_map[33][33] = 34864
+ expected_grid_map[33][39] = 32800
+ expected_grid_map[33][47] = 32800
+ expected_grid_map[33][48] = 32800
+ expected_grid_map[34][5] = 16386
+ expected_grid_map[34][6] = 1025
+ expected_grid_map[34][7] = 1025
+ expected_grid_map[34][8] = 1025
+ expected_grid_map[34][9] = 1025
+ expected_grid_map[34][10] = 33825
+ expected_grid_map[34][11] = 3089
+ expected_grid_map[34][12] = 1025
+ expected_grid_map[34][13] = 1025
+ expected_grid_map[34][14] = 33825
+ expected_grid_map[34][15] = 1025
+ expected_grid_map[34][16] = 1025
+ expected_grid_map[34][17] = 1025
+ expected_grid_map[34][18] = 1025
+ expected_grid_map[34][19] = 1025
+ expected_grid_map[34][20] = 1025
+ expected_grid_map[34][21] = 1025
+ expected_grid_map[34][22] = 1025
+ expected_grid_map[34][23] = 1025
+ expected_grid_map[34][24] = 1025
+ expected_grid_map[34][25] = 2064
+ expected_grid_map[34][33] = 32800
+ expected_grid_map[34][39] = 32800
+ expected_grid_map[34][47] = 32800
+ expected_grid_map[34][48] = 32800
+ expected_grid_map[35][5] = 32800
+ expected_grid_map[35][10] = 32800
+ expected_grid_map[35][14] = 32800
+ expected_grid_map[35][16] = 8192
+ expected_grid_map[35][33] = 32800
+ expected_grid_map[35][39] = 32800
+ expected_grid_map[35][47] = 32800
+ expected_grid_map[35][48] = 32800
+ expected_grid_map[36][5] = 32800
+ expected_grid_map[36][10] = 32800
+ expected_grid_map[36][14] = 32800
+ expected_grid_map[36][16] = 32800
+ expected_grid_map[36][17] = 8192
+ expected_grid_map[36][19] = 8192
+ expected_grid_map[36][33] = 32800
+ expected_grid_map[36][39] = 32800
+ expected_grid_map[36][41] = 8192
+ expected_grid_map[36][47] = 32800
+ expected_grid_map[36][48] = 32800
+ expected_grid_map[37][5] = 32800
+ expected_grid_map[37][10] = 32800
+ expected_grid_map[37][14] = 32800
+ expected_grid_map[37][16] = 32800
+ expected_grid_map[37][17] = 49186
+ expected_grid_map[37][18] = 1025
+ expected_grid_map[37][19] = 2064
+ expected_grid_map[37][33] = 32800
+ expected_grid_map[37][39] = 32800
+ expected_grid_map[37][41] = 32800
+ expected_grid_map[37][42] = 16386
+ expected_grid_map[37][43] = 256
+ expected_grid_map[37][47] = 32800
+ expected_grid_map[37][48] = 32800
+ expected_grid_map[38][5] = 72
+ expected_grid_map[38][6] = 1025
+ expected_grid_map[38][7] = 1025
+ expected_grid_map[38][8] = 1025
+ expected_grid_map[38][9] = 1025
+ expected_grid_map[38][10] = 33825
+ expected_grid_map[38][11] = 1025
+ expected_grid_map[38][12] = 1025
+ expected_grid_map[38][13] = 1025
+ expected_grid_map[38][14] = 33897
+ expected_grid_map[38][15] = 17411
+ expected_grid_map[38][16] = 1097
+ expected_grid_map[38][17] = 38505
+ expected_grid_map[38][18] = 256
+ expected_grid_map[38][33] = 32800
+ expected_grid_map[38][39] = 32800
+ expected_grid_map[38][41] = 32800
+ expected_grid_map[38][42] = 32800
+ expected_grid_map[38][43] = 8192
+ expected_grid_map[38][47] = 32800
+ expected_grid_map[38][48] = 32800
+ expected_grid_map[39][10] = 32800
+ expected_grid_map[39][14] = 32800
+ expected_grid_map[39][15] = 32800
+ expected_grid_map[39][17] = 32800
+ expected_grid_map[39][18] = 4
+ expected_grid_map[39][19] = 4608
+ expected_grid_map[39][33] = 32800
+ expected_grid_map[39][39] = 49186
+ expected_grid_map[39][40] = 17411
+ expected_grid_map[39][41] = 1097
+ expected_grid_map[39][42] = 52275
+ expected_grid_map[39][43] = 3089
+ expected_grid_map[39][44] = 4608
+ expected_grid_map[39][47] = 32800
+ expected_grid_map[39][48] = 32800
+ expected_grid_map[40][10] = 32800
+ expected_grid_map[40][14] = 32800
+ expected_grid_map[40][15] = 128
+ expected_grid_map[40][17] = 32800
+ expected_grid_map[40][18] = 8192
+ expected_grid_map[40][19] = 32800
+ expected_grid_map[40][33] = 32800
+ expected_grid_map[40][39] = 32800
+ expected_grid_map[40][40] = 32800
+ expected_grid_map[40][42] = 32872
+ expected_grid_map[40][43] = 4608
+ expected_grid_map[40][44] = 32800
+ expected_grid_map[40][47] = 32800
+ expected_grid_map[40][48] = 32800
+ expected_grid_map[41][10] = 32800
+ expected_grid_map[41][14] = 32800
+ expected_grid_map[41][17] = 32800
+ expected_grid_map[41][18] = 72
+ expected_grid_map[41][19] = 37408
+ expected_grid_map[41][21] = 8192
+ expected_grid_map[41][33] = 32800
+ expected_grid_map[41][39] = 32800
+ expected_grid_map[41][40] = 128
+ expected_grid_map[41][42] = 32800
+ expected_grid_map[41][43] = 128
+ expected_grid_map[41][44] = 32800
+ expected_grid_map[41][47] = 32800
+ expected_grid_map[41][48] = 32800
+ expected_grid_map[42][10] = 32800
+ expected_grid_map[42][14] = 72
+ expected_grid_map[42][15] = 1025
+ expected_grid_map[42][16] = 1025
+ expected_grid_map[42][17] = 33825
+ expected_grid_map[42][18] = 17411
+ expected_grid_map[42][19] = 52275
+ expected_grid_map[42][20] = 5633
+ expected_grid_map[42][21] = 3089
+ expected_grid_map[42][22] = 1025
+ expected_grid_map[42][23] = 1025
+ expected_grid_map[42][24] = 1025
+ expected_grid_map[42][25] = 1025
+ expected_grid_map[42][26] = 1025
+ expected_grid_map[42][27] = 1025
+ expected_grid_map[42][28] = 1025
+ expected_grid_map[42][29] = 1025
+ expected_grid_map[42][30] = 4608
+ expected_grid_map[42][33] = 32800
+ expected_grid_map[42][39] = 32800
+ expected_grid_map[42][42] = 32800
+ expected_grid_map[42][44] = 32800
+ expected_grid_map[42][47] = 32800
+ expected_grid_map[42][48] = 32800
+ expected_grid_map[43][10] = 32800
+ expected_grid_map[43][17] = 32800
+ expected_grid_map[43][18] = 128
+ expected_grid_map[43][19] = 32800
+ expected_grid_map[43][20] = 32800
+ expected_grid_map[43][30] = 32800
+ expected_grid_map[43][33] = 32800
+ expected_grid_map[43][39] = 32800
+ expected_grid_map[43][42] = 32800
+ expected_grid_map[43][44] = 32800
+ expected_grid_map[43][47] = 32800
+ expected_grid_map[43][48] = 32800
+ expected_grid_map[44][4] = 4
+ expected_grid_map[44][5] = 1025
+ expected_grid_map[44][6] = 1025
+ expected_grid_map[44][7] = 1025
+ expected_grid_map[44][8] = 1025
+ expected_grid_map[44][9] = 1025
+ expected_grid_map[44][10] = 3089
+ expected_grid_map[44][11] = 1025
+ expected_grid_map[44][12] = 1025
+ expected_grid_map[44][13] = 1025
+ expected_grid_map[44][14] = 1025
+ expected_grid_map[44][15] = 1025
+ expected_grid_map[44][16] = 1025
+ expected_grid_map[44][17] = 3089
+ expected_grid_map[44][18] = 1025
+ expected_grid_map[44][19] = 2064
+ expected_grid_map[44][20] = 128
+ expected_grid_map[44][30] = 72
+ expected_grid_map[44][31] = 1025
+ expected_grid_map[44][32] = 1025
+ expected_grid_map[44][33] = 35889
+ expected_grid_map[44][34] = 1025
+ expected_grid_map[44][35] = 1025
+ expected_grid_map[44][36] = 1025
+ expected_grid_map[44][37] = 1025
+ expected_grid_map[44][38] = 1025
+ expected_grid_map[44][39] = 33825
+ expected_grid_map[44][40] = 1025
+ expected_grid_map[44][41] = 1025
+ expected_grid_map[44][42] = 2064
+ expected_grid_map[44][44] = 32800
+ expected_grid_map[44][47] = 32800
+ expected_grid_map[44][48] = 32800
+ expected_grid_map[45][33] = 32872
+ expected_grid_map[45][34] = 1025
+ expected_grid_map[45][35] = 1025
+ expected_grid_map[45][36] = 1025
+ expected_grid_map[45][37] = 1025
+ expected_grid_map[45][38] = 1025
+ expected_grid_map[45][39] = 33825
+ expected_grid_map[45][40] = 1025
+ expected_grid_map[45][41] = 1025
+ expected_grid_map[45][42] = 1025
+ expected_grid_map[45][43] = 1025
+ expected_grid_map[45][44] = 1097
+ expected_grid_map[45][45] = 1025
+ expected_grid_map[45][46] = 1025
+ expected_grid_map[45][47] = 34864
+ expected_grid_map[45][48] = 32800
+ expected_grid_map[46][33] = 32800
+ expected_grid_map[46][39] = 32800
+ expected_grid_map[46][47] = 32800
+ expected_grid_map[46][48] = 32800
+ expected_grid_map[47][33] = 32800
+ expected_grid_map[47][39] = 32800
+ expected_grid_map[47][47] = 32800
+ expected_grid_map[47][48] = 128
+ expected_grid_map[48][33] = 32800
+ expected_grid_map[48][39] = 32800
+ expected_grid_map[48][47] = 32800
+ expected_grid_map[49][33] = 72
+ expected_grid_map[49][34] = 1025
+ expected_grid_map[49][35] = 1025
+ expected_grid_map[49][36] = 1025
+ expected_grid_map[49][37] = 1025
+ expected_grid_map[49][38] = 1025
+ expected_grid_map[49][39] = 2136
+ expected_grid_map[49][40] = 1025
+ expected_grid_map[49][41] = 1025
+ expected_grid_map[49][42] = 1025
+ expected_grid_map[49][43] = 1025
+ expected_grid_map[49][44] = 1025
+ expected_grid_map[49][45] = 1025
+ expected_grid_map[49][46] = 1025
+ expected_grid_map[49][47] = 2064
- # approximative test (replace image comparison)
- assert (np.sum(env.rail.grid) == 12054975)
- print(env.rail.grid)
+ assert np.array_equal(env.rail.grid, expected_grid_map), "actual={}, expected={}".format(env.rail.grid,
+ expected_grid_map)
s0 = 0
s1 = 0
for a in range(env.get_num_agents()):