def testGetStateScope_AlreadyUsedNameScope(self):
   with tf.Graph().as_default():
     with tf.compat.v1.name_scope("a"):
       pass
     with self.assertRaisesRegexp(RuntimeError,
                                  "name_scope was already taken"):
       module._try_get_state_scope("a", False)
示例#2
0
    def testGetStateScope_UnusedNameScope(self):
        self.assertEqual(module._try_get_state_scope("a", False), "a/")
        with tf.name_scope("a") as ns:
            self.assertEqual(ns, "a/")

        self.assertEqual(module._try_get_state_scope("a", False), "a_1/")
        with tf.name_scope("a") as ns:
            self.assertEqual(ns, "a_1/")
示例#3
0
文件: module_test.py 项目: jankim/hub
  def testGetStateScope_UnusedNameScope(self):
    self.assertEqual(module._try_get_state_scope("a", False), "a/")
    with tf.name_scope("a") as ns:
      self.assertEqual(ns, "a/")

    self.assertEqual(module._try_get_state_scope("a", False), "a_1/")
    with tf.name_scope("a") as ns:
      self.assertEqual(ns, "a_1/")
  def testGetStateScope_UnusedNameScope(self):
    with tf.Graph().as_default():
      self.assertEqual(module._try_get_state_scope("a", False), "a/")
      with tf.compat.v1.name_scope("a") as ns:
        self.assertEqual(ns, "a/")

      self.assertEqual(module._try_get_state_scope("a", False), "a_1/")
      with tf.compat.v1.name_scope("a") as ns:
        self.assertEqual(ns, "a_1/")
  def testGetStateScopeWithActiveScopes(self):
    with tf.Graph().as_default():
      with tf.compat.v1.name_scope("foo"):
        abs_scope = module._try_get_state_scope("a", False)
        self.assertEqual(abs_scope, "a/")
        with tf.compat.v1.name_scope(abs_scope) as ns:
          self.assertEqual(ns, "a/")

    with tf.Graph().as_default():
      with tf.compat.v1.variable_scope("vs"):
        self.assertEqual(module._try_get_state_scope("a", False), "vs/a/")
        with tf.compat.v1.name_scope(name="a") as ns:
          self.assertEqual(ns, "vs/a/")

    with tf.Graph().as_default():
      with tf.compat.v1.name_scope("foo"):
        with tf.compat.v1.variable_scope("vs"):
          self.assertEquals(module._try_get_state_scope("a", False), "vs/a/")
示例#6
0
文件: module_test.py 项目: jankim/hub
  def testGetStateScopeWithActiveScopes(self):
    with tf.Graph().as_default():
      with tf.name_scope("foo"):
        abs_scope = module._try_get_state_scope("a", False)
        self.assertEqual(abs_scope, "a/")
        with tf.name_scope(abs_scope) as ns:
          self.assertEqual(ns, "a/")

    with tf.Graph().as_default():
      with tf.variable_scope("vs"):
        self.assertEqual(module._try_get_state_scope("a", False), "vs/a/")
        with tf.name_scope(name="a") as ns:
          self.assertEqual(ns, "vs/a/")

    with tf.Graph().as_default():
      with tf.name_scope("foo"):
        with tf.variable_scope("vs"):
          self.assertEquals(module._try_get_state_scope("a", False), "vs/a/")
 def testGetStateScope_UsesVariableScope(self):
   with tf.Graph().as_default():
     self.assertEqual(module._try_get_state_scope("a"), "a/")
     with tf.compat.v1.variable_scope(None, default_name="a") as vs:
       self.assertEqual(vs.name, "a_1")
 def testGetStateScope(self):
   with tf.Graph().as_default():
     self.assertEqual(module._try_get_state_scope("a"), "a/")
     self.assertEqual(module._try_get_state_scope("a"), "a_1/")
示例#9
0
 def testGetStateScope(self):
     self.assertEqual(module._try_get_state_scope("a"), "a/")
     self.assertEqual(module._try_get_state_scope("a"), "a_1/")
示例#10
0
 def testGetStateScope_UsesNameScope(self):
     with tf.Graph().as_default():
         self.assertEqual(module._try_get_state_scope("a"), "a/")
         with tf_v1.name_scope("a") as ns:
             self.assertEqual(ns, "a_1/")
示例#11
0
文件: module_test.py 项目: jankim/hub
 def testGetStateScope_AlreadyUsedNameScope(self):
   with tf.name_scope("a"):
     pass
   with self.assertRaisesRegexp(RuntimeError, "name_scope was already taken"):
     module._try_get_state_scope("a", False)
示例#12
0
文件: module_test.py 项目: jankim/hub
 def testGetStateScope_UsesVariableScope(self):
   self.assertEqual(module._try_get_state_scope("a"), "a/")
   with tf.variable_scope(None, default_name="a") as vs:
     self.assertEqual(vs.name, "a_1")
示例#13
0
文件: module_test.py 项目: jankim/hub
 def testGetStateScope(self):
   self.assertEqual(module._try_get_state_scope("a"), "a/")
   self.assertEqual(module._try_get_state_scope("a"), "a_1/")