This article gives an outline of the progress of work on the Study of UWB (University of West Bohemia) test bench for partly-suspended and fullysuspended variants of individual wheelset traction drives. First, a survey of existing designs of roller test benches was conducted. Its outcomes supported the idea of designing a completely new device. The main part of the article is devoted to the design of the test bench. Furthermore, the article briefly summarizes the progress and especially the partial results of the work on the task Computer simulation for optimization of the structural arrangement of the traction drive, which focuses on the development of a mathematical model of the aforementioned test bench.