USAFA did not have a similar test. USAFA has a number of 'flying' opportunities with the glider program, the parachute program, powered flight etc. My impression has always been that USAFA puts you in the air to determine if you have the basic skill set to learn how to become a pilot.
I think the OP is talking about the AFOQT and TBAS, but wanted to make a note here-
Honestly, USAFA doesn't really screen people for aptitude for flying in its basic airmanship programs. All airmanship programs are pass/fail and don't affect whether or not you can get a pilot slot...with the exception of Powered Flight. They've made Powered Flight essentially mandatory in recent years for pilot selection (you can still get a waiver if for some reason you can't get in the program), but pretty much as long as you show up to the airfield for Powered Flight, you'll pass. They don't really have any skill requirement to complete the program, and even if you develop airsickness or are really terrible at flying, they're just trying to give you a brief exposure to what flying a single engine aircraft is like.
The first time you're really screened for flying aptitude is when you start training after graduation in Pueblo (IFT, but they're changing back to IFS) or at your UPT base.
The rationale there is that USAFA airmanship programs should motivate cadets to pursue rated careers and develop big picture airmaship/leadership skills, not serve as screeners. Another nuance there is that for cadets interested in applying for ENJJPT, participation in one of the advanced airmanship programs (WOB, Soaring IPs, Flying Team) can provide an advantage in the form of a recommendation from an airfield squadron commander.
Long post, but I just wanted to make the point that you aren't truly screened for flying aptitude until you're a lieutenant in AETC.