def test_add_inmate(self): Inmate_TestDouble.clear_class_vars() inmates = Inmates(Inmate_TestDouble, Mock()) inmate_details = Mock() inmate_id = 23 inmate_details.jail_id.return_value = inmate_id inmates.add(inmate_id, inmate_details) assert Inmate_TestDouble.instantiated_called(1) inmate = Inmate_TestDouble.instantiated[0] assert inmate.inmate_details == inmate_details assert inmate.saved_count == 1