def test_args_variational(self): args = VariationalArgs() self.assertTrue(True) args = VariationalArgs(output_samples=1) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('method=variational', ' '.join(cmd)) self.assertIn('output_samples=1', ' '.join(cmd)) args = VariationalArgs(tol_rel_obj=0.01) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('method=variational', ' '.join(cmd)) self.assertIn('tol_rel_obj=0.01', ' '.join(cmd)) args = VariationalArgs(adapt_engaged=True, adapt_iter=100) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('adapt engaged=1 iter=100', ' '.join(cmd)) args = VariationalArgs(adapt_engaged=False) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('adapt engaged=0', ' '.join(cmd)) args = VariationalArgs(eta=0.1) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('eta=0.1', ' '.join(cmd))
def test_args_variational(self): args = VariationalArgs() self.assertTrue(True) args = VariationalArgs(output_samples=1) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('method=variational', ' '.join(cmd)) self.assertIn('output_samples=1', ' '.join(cmd)) args = VariationalArgs(tol_rel_obj=1) args.validate(chains=1) cmd = args.compose(idx=0, cmd=[]) self.assertIn('method=variational', ' '.join(cmd)) self.assertIn('tol_rel_obj=1', ' '.join(cmd))