`tests/test_dvae_winci2020.py` is too bulky

[kevinchern]

test_dvae_winci2020.py dominates a bulk of test time (about 93% of 7 seconds).
The unit tests should be designed to test the logic without a full training loop.