def test_image_shape_to_grids(self):

    def graph_fn():
        (y_grid, x_grid) = ta_utils.image_shape_to_grids(height=2, width=3)
        return (y_grid, x_grid)
    expected_y_grid = np.array([[0, 0, 0], [1, 1, 1]])
    expected_x_grid = np.array([[0, 1, 2], [0, 1, 2]])
    (y_grid, x_grid) = self.execute(graph_fn, [])
    np.testing.assert_array_equal(y_grid, expected_y_grid)
    np.testing.assert_array_equal(x_grid, expected_x_grid)
def test_image_shape_to_grids(self):

    def graph_fn():
        (y_grid, x_grid) = ta_utils.image_shape_to_grids(height=2, width=3)
        return (y_grid, x_grid)
    expected_y_grid = np.array([[0, 0, 0], [1, 1, 1]])
    expected_x_grid = np.array([[0, 1, 2], [0, 1, 2]])
    (y_grid, x_grid) = self.execute(graph_fn, [])
    np.testing.assert_array_equal(y_grid, expected_y_grid)
    np.testing.assert_array_equal(x_grid, expected_x_grid)