Have you tried Anki's image occlusion and Cloze deletion feature. You can fit entire diagrams or texts that give you the "whole picture" all the while blanking out certain portions of it to test yourself. Anki is great. Admittedly, basic flashcards do have their limitations.