def test_randomize_weights(self):
    m = Model()
    input = torch.randn(5)
    normal1 = run_model(m, input, False)
    normal2 = run_model(m, input, False)
    rand = run_model(m, input, True)
    assert torch.allclose(normal1, normal2)
    assert (not torch.allclose(normal1, rand))
def test_randomize_weights(self):
    m = Model()
    input = torch.randn(5)
    normal1 = run_model(m, input, False)
    normal2 = run_model(m, input, False)
    rand = run_model(m, input, True)
    assert torch.allclose(normal1, normal2)
    assert (not torch.allclose(normal1, rand))