Use assert_array_less

Author: toslunar

tests/functions_tests/test_invert_gradients.py

@@ -53,12 +53,12 @@ def check_forward(self, x_data):
 
         loss = functions.sum(y)  # Minimize y
         loss.backward()
-        self.assertTrue((x.grad > 0).all())  # Decrease x
+        numpy.testing.assert_array_less(0, x.grad)  # Decrease x
         x.cleargrad()
 
         loss = -functions.sum(y)  # Maximize y
         loss.backward()
-        self.assertTrue((x.grad > 0).all())  # Decrease x
+        numpy.testing.assert_array_less(0, x.grad)  # Decrease x
         x.cleargrad()
 
         # Exceeding range_min
@@ -68,12 +68,12 @@ def check_forward(self, x_data):
 
         loss = functions.sum(y)  # Minimize y
         loss.backward()
-        self.assertTrue((x.grad < 0).all())  # Increase x
+        numpy.testing.assert_array_less(x.grad, 0)  # Increase x
         x.cleargrad()
 
         loss = -functions.sum(y)  # Maximize y
         loss.backward()
-        self.assertTrue((x.grad < 0).all())  # Increase x
+        numpy.testing.assert_array_less(x.grad, 0)  # Increase x
         x.cleargrad()
 
     @condition.retry(3)