I was just playing with my Riptide trident during rain. I went about y 200 with trident and then kept going forward. I then smoothly came down but still I died. Then I ran a couple of tests and realised that tridents don't reset your fall if used in mid-air..... I have attached a video please consider it too. Mind the enchantments I am using too!