Skip to content

Commit b862de2

Browse files
committed
Uses absltest rather than googletest.
PiperOrigin-RevId: 317982291
1 parent 0c5f0ea commit b862de2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_data_validation/statistics/generators/natural_language_stats_generator_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,12 @@
1818

1919
from __future__ import print_function
2020

21+
from absl.testing import absltest
2122
import pyarrow as pa
2223
from tensorflow_data_validation.statistics.generators import natural_language_stats_generator as nlsg
2324
from tensorflow_data_validation.utils import test_util
2425
from typing import Text
2526

26-
import unittest
2727
from tensorflow_metadata.proto.v0 import statistics_pb2
2828

2929

@@ -203,4 +203,4 @@ def test_nl_generator_avg_word_heuristic_non_match(self):
203203

204204

205205
if __name__ == '__main__':
206-
unittest.main()
206+
absltest.main()

0 commit comments

Comments
 (0)