sped up some tests. still need to speed up test_save_load by getting rid of more inference