def test_evaluate(): diag_lv = DiagLazyVariable(Variable(diag)) assert torch.equal(diag_lv.evaluate().data, diag.diag())