83fd1cbd35
* Previously, strong assumptions were made about how numpy.ndarrays are formatted as strings. This led to breakages due to certain unclear changes in numpy or its dependencies. This CL relaxes the assumption and fix the affected tests for tfdbg and eager. * The tests in tensor_format_test.py are simplified through helper methods. PiperOrigin-RevId: 181494182
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
# Copyright 2018 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
"""Testing utilities for tfdbg command-line interface."""
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import re
|
|
|
|
import numpy as np
|
|
|
|
|
|
def assert_lines_equal_ignoring_whitespace(test, expected_lines, actual_lines):
|
|
"""Assert equality in lines, ignoring all whitespace.
|
|
|
|
Args:
|
|
test: An instance of unittest.TestCase or its subtypes (e.g.,
|
|
TensorFlowTestCase).
|
|
expected_lines: Expected lines as an iterable of strings.
|
|
actual_lines: Actual lines as an iterable of strings.
|
|
"""
|
|
test.assertEqual(
|
|
len(expected_lines), len(actual_lines),
|
|
"Mismatch in the number of lines: %d vs %d" % (
|
|
len(expected_lines), len(actual_lines)))
|
|
for expected_line, actual_line in zip(expected_lines, actual_lines):
|
|
test.assertEqual("".join(expected_line.split()),
|
|
"".join(actual_line.split()))
|
|
|
|
|
|
# Regular expression for separators between values in a string representation
|
|
# of an ndarray, exclusing whitespace.
|
|
_ARRAY_VALUE_SEPARATOR_REGEX = re.compile(r"(array|\(|\[|\]|\)|\||,)")
|
|
|
|
|
|
def assert_array_lines_close(test, expected_array, array_lines):
|
|
"""Assert that the array value represented by lines is close to expected.
|
|
|
|
Note that the shape of the array represented by the `array_lines` is ignored.
|
|
|
|
Args:
|
|
test: An instance of TensorFlowTestCase.
|
|
expected_array: Expected value of the array.
|
|
array_lines: A list of strings representing the array.
|
|
E.g., "array([[ 1.0, 2.0 ], [ 3.0, 4.0 ]])"
|
|
Assumes that values are separated by commas, parentheses, brackets, "|"
|
|
characters and whitespace.
|
|
"""
|
|
elements = []
|
|
for line in array_lines:
|
|
line = re.sub(_ARRAY_VALUE_SEPARATOR_REGEX, " ", line)
|
|
elements.extend(float(s) for s in line.split())
|
|
test.assertAllClose(np.array(expected_array).flatten(), elements)
|