def test_parameter_setting(param, expected): vocab = [] mod = TorchRNNClassifier(vocab) mod.set_params(**{param: expected}) result = getattr(mod, param) if param == "embedding": assert np.array_equal(result, expected) else: assert result == expected
def test_np_parameter_setting(param, expected): vocab = [] mod = TorchRNNClassifier(vocab) mod.set_params(**{param: expected}) result = getattr(mod, param) assert result == expected