aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py')
-rw-r--r--tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py303
1 files changed, 0 insertions, 303 deletions
diff --git a/tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py b/tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py
deleted file mode 100644
index 428ebbedca..0000000000
--- a/tensorflow/contrib/autograph/pyct/static_analysis/cfg_test.py
+++ /dev/null
@@ -1,303 +0,0 @@
-# Copyright 2017 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.
-# ==============================================================================
-"""Tests for cfg module."""
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
-import functools
-
-import gast
-
-from tensorflow.contrib.autograph.pyct import anno
-from tensorflow.contrib.autograph.pyct import parser
-from tensorflow.contrib.autograph.pyct import qual_names
-from tensorflow.contrib.autograph.pyct import transformer
-from tensorflow.contrib.autograph.pyct.static_analysis import cfg
-from tensorflow.python.platform import test
-
-
-class CFGTest(test.TestCase):
-
- def _parse_and_analyze(self, test_fn):
- node, source = parser.parse_entity(test_fn)
- entity_info = transformer.EntityInfo(
- source_code=source,
- source_file=None,
- namespace={},
- arg_values=None,
- arg_types=None,
- owner_type=None)
- node = qual_names.resolve(node)
- return node, entity_info
-
- def _check_anno_matches(self, node, anno_name, var_names):
- if isinstance(var_names, str):
- var_names = (var_names,)
- qual_vars = set()
- for var_name in var_names:
- if isinstance(var_name, str):
- if '[' in var_name or ']' in var_name:
- raise ValueError('Annotation matching not supported with subscript.')
- if '.' not in var_name:
- qual_vars.add(qual_names.QN(var_name))
- else:
- attrs = var_name.split('.')
- this_qn = functools.reduce(qual_names.QN, attrs[1:],
- qual_names.QN(attrs[0]))
- qual_vars.add(this_qn)
- self.assertEqual(anno.getanno(node, anno_name), qual_vars)
-
- def test_reaching(self):
-
- def f(x):
- print(x)
- while True:
- x = x
- x = x
- return x
-
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.ReachingDefinitions(ctx))
- body = node.body[0].body
- # Only the argument reaches the expression
- def_in = anno.getanno(body[0], 'definitions_in')
- # One element, x, from arguments
- self.assertEqual(set(type(d[1]) for d in def_in), set((gast.arguments,)))
-
- while_body = body[1].body
- def_in = anno.getanno(while_body[0], 'definitions_in')
- # One definition, two possible sources.
- # - One from an assignment (if the loop is entered)
- # - The other from the arguments (if loop is not entered)
- self.assertEqual(
- set(type(d[1]) for d in def_in), set((gast.arguments, gast.Assign)))
-
- def_in = anno.getanno(while_body[1], 'definitions_in')
- # If we've reached this line, the only reaching definition of x is the
- # Assign node in previous line
- self.assertEqual(set(type(d[1]) for d in def_in), set((gast.Assign,)))
-
- def_in = anno.getanno(body[2], 'definitions_in')
- # Same situation as while_body[0]
- self.assertEqual(
- set(type(d[1]) for d in def_in), set((gast.arguments, gast.Assign)))
-
- def test_defined(self):
-
- def f(x):
- if x:
- y = 2 # pylint: disable=unused-variable
- return x
-
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.Defined(ctx))
- body = node.body[0].body
- # only x is for sure defined at the end
- self._check_anno_matches(body[1], 'defined_in', 'x')
- # at the end of the if body both x and y are defined
- if_body = body[0].body
- self._check_anno_matches(if_body[0], 'defined_out', ('x', 'y'))
-
- def _get_live_annotated_fnbody(self, f):
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.Liveness(ctx))
- body = node.body[0].body
- return body
-
- def test_live_straightline(self):
-
- def f1(x):
- a = g(x) # pylint: disable=undefined-variable
- b = h(a) # pylint: disable=undefined-variable, unused-variable
- return x
-
- body = self._get_live_annotated_fnbody(f1)
- self._check_anno_matches(body[1], 'live_in', ('a', 'h', 'x'))
- self._check_anno_matches(body[2], 'live_in', ('x'))
- self._check_anno_matches(body[0], 'live_in', ('g', 'h', 'x'))
- self._check_anno_matches(body[2], 'live_out', ())
-
- def test_live_stacked_conds_with_else(self):
-
- def f2(x, a): # pylint: disable=unused-argument
- if a > 0: # x should not be live
- x = 0
- if a > 1:
- x = 1
- else:
- x = 2
-
- body = self._get_live_annotated_fnbody(f2)
- self._check_anno_matches(body[0], 'live_in', ('a'))
- self._check_anno_matches(body[1], 'live_in', ('a'))
-
- def test_live_stacked_conds(self):
-
- def f3(x, a):
- if a > 0: # x and a should be live
- x = 0
- if a > 1: # x and a should be live_in
- x = 1
- return x # x should be live
-
- body = self._get_live_annotated_fnbody(f3)
- self._check_anno_matches(body[0], 'live_in', ('a', 'x'))
- self._check_anno_matches(body[1], 'live_in', ('a', 'x'))
- self._check_anno_matches(body[2], 'live_in', ('x'))
-
- def test_live_possibly_unused_cond(self):
-
- def f4(x, a):
- if a > 0: # x should be live
- x = 0
- x += 1
-
- body = self._get_live_annotated_fnbody(f4)
- self._check_anno_matches(body[0], 'live_in', ('x', 'a'))
- self._check_anno_matches(body[1], 'live_in', ('x'))
-
- def test_live_attribute_in_cond(self):
-
- def f5(x, a):
- if a > 0: # x.y should be live
- x.y = 0
- return x.y
-
- body = self._get_live_annotated_fnbody(f5)
- self._check_anno_matches(body[0], 'live_in', ('x', 'x.y', 'a'))
-
- def test_live_noop(self):
-
- def f6(x):
- return x # should this cause x.* to be live?
-
- body = self._get_live_annotated_fnbody(f6)
- self._check_anno_matches(body[0], 'live_in', ('x'))
-
- def test_live_loop(self):
-
- def f7(x, n):
- for i in range(n):
- x += i
- return x
-
- body = self._get_live_annotated_fnbody(f7)
- self._check_anno_matches(body[0], 'live_in', ('x', 'n', 'range'))
- self._check_anno_matches(body[1], 'live_in', ('x'))
-
- def test_live_context_manager(self):
-
- def f8(x, f):
- with f:
- x += 1
-
- body = self._get_live_annotated_fnbody(f8)
- self._check_anno_matches(body[0], 'live_in', ('f', 'x'))
-
- def test_node_equality(self):
- node_a = gast.parse('y = x').body[0]
- node_b = gast.parse('y = x').body[0]
- self.assertNotEqual(node_a, node_b)
-
- def test_nested_functions_defined(self):
-
- def f(x):
- y = x * 2
-
- def g(z):
- return z + y
-
- return g(x)
-
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.Defined(ctx))
-
- body = node.body[0].body
- self.assertEqual(
- anno.getanno(body[2], 'defined_in'),
- frozenset(map(qual_names.QN, ('g', 'x', 'y'))))
-
- # TODO(alexbw): CFG analysis doesn't currently cross FunctionDef boundaries.
- # NOTE: 'z' is easy to find, but 'y' is not identified as
- # defined, because CFG analysis is applied with each function separately.
- # fndef_body = body[1].body
- # self.assertEqual(
- # anno.getanno(fndef_body[0], 'defined_in'),
- # frozenset(map(qual_names.QN, ('z', 'y'))))
-
- def test_nested_functions_dont_leak_definitions(self):
-
- def f(x):
- print(x)
-
- def g():
- y = 2
- return y
-
- return g() # y is not defined here
-
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.Defined(ctx))
- body = node.body[0].body
- self.assertEqual(
- anno.getanno(body[2], 'defined_in'),
- frozenset(map(qual_names.QN, ('x', 'g'))))
-
- def test_loop_else(self):
-
- # Disabling useless-else-on-loop error, because 'break' and 'continue'
- # canonicalization are a separate analysis pass, and here we test
- # the CFG analysis in isolation.
- def for_orelse(x):
- y = 0
- for i in range(len(x)):
- x += i
- else: # pylint: disable=useless-else-on-loop
- y = 1
- return x, y
-
- def while_orelse(x, i):
- y = 0
- while x < 10:
- x += i
- else: # pylint: disable=useless-else-on-loop
- y = 1
- return x, y
-
- for f in (for_orelse, while_orelse):
- node, ctx = self._parse_and_analyze(f)
- cfg.run_analyses(node, cfg.ReachingDefinitions(ctx))
- body = node.body[0].body
- return_node = body[-1]
- reaching_defs = anno.getanno(return_node, 'definitions_in')
-
- # Y could be defined by Assign(Num(0)) or Assign(Num(1))
- # X could be defined as an argument or an AugAssign.
- y_defs = [node for var, node in reaching_defs if str(var) == 'y']
- x_defs = [node for var, node in reaching_defs if str(var) == 'x']
-
- self.assertEqual(set((gast.Assign,)), set(type(def_) for def_ in y_defs))
- self.assertEqual(set((0, 1)), set(def_.value.n for def_ in y_defs))
- self.assertEqual(len(y_defs), 2)
- self.assertEqual(
- set((gast.arguments, gast.AugAssign)),
- set(type(def_) for def_ in x_defs))
- self.assertEqual(len(x_defs), 2)
-
-
-if __name__ == '__main__':
- test.main()
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-04 19:57:44 +0000