Browse Source

test(mge/test): reorganize utils test files

GitOrigin-RevId: 08d27a8b1d
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
bfb9def988
6 changed files with 3 additions and 0 deletions
  1. +0
    -0
      imperative/python/test/unit/core/test_zero_dim_tensor.py
  2. +3
    -0
      imperative/python/test/unit/jit/test_tracing.py
  3. +0
    -0
      imperative/python/test/unit/utils/test_cgtools.py
  4. +0
    -0
      imperative/python/test/unit/utils/test_dump_naming.py
  5. +0
    -0
      imperative/python/test/unit/utils/test_network_node.py
  6. +0
    -0
      imperative/python/test/unit/utils/test_utils.py

imperative/python/test/unit/test_zero_dim_tensor.py → imperative/python/test/unit/core/test_zero_dim_tensor.py View File


imperative/python/test/unit/test_tracing.py → imperative/python/test/unit/jit/test_tracing.py View File

@@ -28,6 +28,7 @@ from megengine.functional import exp, log
from megengine.jit import exclude_from_trace, trace from megengine.jit import exclude_from_trace, trace
from megengine.module import Module from megengine.module import Module
from megengine.random import normal, uniform from megengine.random import normal, uniform
from megengine.utils.naming import auto_naming




@pytest.mark.parametrize("trace_mode", [False, True]) @pytest.mark.parametrize("trace_mode", [False, True])
@@ -139,6 +140,8 @@ def test_dump():
def f(a, b): def f(a, b):
return a + b return a + b


# prevent from remaining scope from exception test
auto_naming.clear()
a = tensor([2]) a = tensor([2])
b = tensor([4]) b = tensor([4])
y = f(a, b).numpy() y = f(a, b).numpy()

imperative/python/test/unit/test_cgtools.py → imperative/python/test/unit/utils/test_cgtools.py View File


imperative/python/test/unit/test_dump_naming.py → imperative/python/test/unit/utils/test_dump_naming.py View File


imperative/python/test/unit/utils/test_opr.py → imperative/python/test/unit/utils/test_network_node.py View File


imperative/python/test/unit/test_utils.py → imperative/python/test/unit/utils/test_utils.py View File


Loading…
Cancel
Save