Split url jQuery

 // show block on test or test2 page
        $(window).load(function() {

          var segments = window.location.pathname.split( '/')
          if (segments.length > 3) {
            $('.block').hide();
          }

          if (segments[2] == "TEST" || segments[2] == "TEST2"){
            $('.block').show();

          }
        });