| 466 | return self.evaluate(tf_ans) |
| 467 | |
| 468 | def testBasic(self): |
| 469 | self.assertTrue( |
| 470 | np.array_equal(self._Range(0, 5, 1), np.array([0, 1, 2, 3, 4]))) |
| 471 | self.assertTrue(np.array_equal(self._Range(0, 5, 2), np.array([0, 2, 4]))) |
| 472 | self.assertTrue(np.array_equal(self._Range(0, 6, 2), np.array([0, 2, 4]))) |
| 473 | self.assertTrue( |
| 474 | np.array_equal(self._Range(13, 32, 7), np.array([13, 20, 27]))) |
| 475 | self.assertTrue( |
| 476 | np.array_equal( |
| 477 | self._Range(100, 500, 100), np.array([100, 200, 300, 400]))) |
| 478 | self.assertEqual(math_ops.range(0, 5, 1).dtype, dtypes.int32) |
| 479 | |
| 480 | @test_util.run_deprecated_v1 |
| 481 | def testLimitOnly(self): |