def test_partial_function(self): clip_between = partial(np.clip, a_min=1, a_max=4) self._test(clip_between) array = np.arange(10000, dtype=np.float32) actual = Elementwise(array, None, clip_between) assert_array_equal(actual.ndarray(), clip_between(array)) self.assertEqual(actual.ndarray().max(), 4)
def test_partial_function(self): clip_between = partial(np.clip, a_min=1, a_max=4) self._test(clip_between) array = np.arange(10000, dtype=np.float32) actual = Elementwise(array, None, clip_between) assert_array_equal(actual.ndarray(), clip_between(array)) self.assertEqual(actual.ndarray().max(), 4)