You can hide any of the respective gridlines using the following lines:

test_ax.xaxis._axinfo['grid'].update({'linewidth': 0}) 
test_ax.yaxis._axinfo['grid'].update({'linewidth': 0})
test_ax.zaxis._axinfo['grid'].update({'linewidth': 0})