I know you didn't create this test, but I think it would be better to manually create the trace, that way you know what you expect the results to be. Something like the following

```
model = simple_model()[0]
trace = NpTrace(model.vars)

for v in [1,2,1,2]: 
    trace.record({'x' : v}) 
```