test

wandb · Aug 24, 2024 · 434d819 · 434d819
1 parent a5032ba
commit 434d819
Show file tree

Hide file tree

Showing 13 changed files with 16 additions and 16 deletions.
diff --git a/weave/legacy/tests/test_arrow.py b/weave/legacy/tests/test_arrow.py
@@ -37,8 +37,8 @@
 from weave.legacy.weave.ops_primitives import list_, make_list
 from .util import list_arrow_test_helpers as lath
 
-from weave.legacy.weave.tests.util import tag_test_util as ttu
-from weave.legacy.weave.tests.util import weavejs_ops
+from weave.legacy.tests.util import tag_test_util as ttu
+from weave.legacy.tests.util import weavejs_ops
 from . import test_wb
 
 _loading_builtins_token = context_state.set_loading_built_ins()

diff --git a/weave/legacy/tests/test_arrow_awl.py b/weave/legacy/tests/test_arrow_awl.py
@@ -8,7 +8,7 @@
 from weave.legacy.weave.arrow.convert import to_arrow
 from weave.legacy.weave.language_features.tagging import tag_store, tagged_value_type
 
-from weave.legacy.weave.tests.util.concrete_tagged_value import (
+from weave.legacy.tests.util.concrete_tagged_value import (
     TaggedValue,
     concrete_from_tagstore,
     concrete_to_tagstore,

diff --git a/weave/legacy/tests/test_arrow_topy.py b/weave/legacy/tests/test_arrow_topy.py
@@ -6,7 +6,7 @@
 import weave
 from weave.legacy.weave import ops_arrow
 
-from weave.legacy.weave.tests.util.concrete_tagged_value import (
+from weave.legacy.tests.util.concrete_tagged_value import (
     TaggedValue,
     concrete_from_tagstore,
     concrete_to_tagstore,

diff --git a/weave/legacy/tests/test_custom_types.py b/weave/legacy/tests/test_custom_types.py
@@ -6,7 +6,7 @@
 from weave.legacy.weave import ops_arrow
 
 from ... import errors
-from weave.legacy.weave.tests.util import geom
+from weave.legacy.tests.util import geom
 
 
 def test_mapped_method_on_custom_type():

diff --git a/weave/legacy/tests/test_execute_fast.py b/weave/legacy/tests/test_execute_fast.py
@@ -2,7 +2,7 @@
 from weave.legacy.weave import dispatch, weave_internal
 from weave.legacy.weave import weave_types as types
 
-from weave.legacy.weave.tests.util import weavejs_ops
+from weave.legacy.tests.util import weavejs_ops
 
 
 def test_nested_weavejs_call():

diff --git a/weave/legacy/tests/test_list_arrow_compat.py b/weave/legacy/tests/test_list_arrow_compat.py
@@ -14,8 +14,8 @@
 )
 from weave.legacy.weave.ops_primitives import dict_, list_
 
-from weave.legacy.weave.tests.util import tag_test_util as ttu
-from weave.legacy.weave.tests.util import list_arrow_test_helpers as lath
+from weave.legacy.tests.util import tag_test_util as ttu
+from weave.legacy.tests.util import list_arrow_test_helpers as lath
 
 
 def filter_fn(row) -> bool:

diff --git a/weave/legacy/tests/test_op_behaviors.py b/weave/legacy/tests/test_op_behaviors.py
@@ -28,12 +28,12 @@
 )
 
 from ...legacy import registry_mem
-from weave.legacy.weave.tests.util.concrete_tagged_value import (
+from weave.legacy.tests.util.concrete_tagged_value import (
     TaggedValue,
     concrete_from_tagstore,
     concrete_to_tagstore,
 )
-from weave.legacy.weave.tests.util.op_specs import OP_TEST_SPECS, OpSpec, OpSpecTestCase
+from weave.legacy.tests.util.op_specs import OP_TEST_SPECS, OpSpec, OpSpecTestCase
 
 
 def assert_equal_with_tags(node: graph.Node, v: typing.Any, expected: typing.Any):

diff --git a/weave/legacy/tests/test_table_ops.py b/weave/legacy/tests/test_table_ops.py
@@ -16,7 +16,7 @@
 from weave.legacy.weave import weave_types as types
 from weave.legacy.weave.ops_domain import table as table_ops
 
-from weave.legacy.weave.tests.util import weavejs_ops
+from weave.legacy.tests.util import weavejs_ops
 
 TABLE_TYPES = ["list", "pandas", "sql"]
 

diff --git a/weave/legacy/tests/test_wb.py b/weave/legacy/tests/test_wb.py
@@ -20,7 +20,7 @@
 from .test_wb_domain_ops import assert_gql_str_equal
 
 from ...tests import fixture_fakewandb as fwb
-from weave.legacy.weave.tests.util import weavejs_ops
+from weave.legacy.tests.util import weavejs_ops
 
 file_path_response = {
     "project_518fa79465d8ffaeb91015dce87e092f": {

diff --git a/weave/legacy/tests/test_weavify.py b/weave/legacy/tests/test_weavify.py
@@ -4,7 +4,7 @@
 from weave.legacy.weave import weave_types as types
 
 from ...legacy import weavify
-from weave.legacy.weave.tests.util import geom
+from weave.legacy.tests.util import geom
 
 
 @pytest.mark.parametrize(

diff --git a/weave/legacy/weave/ops_primitives/test_list.py b/weave/legacy/weave/ops_primitives/test_list.py
@@ -10,7 +10,7 @@
     tagged_value_type,
 )
 from weave.legacy.weave.ops_primitives import dict, errors, list_, number, runs
-from weave.legacy.weave.tests.util import geom, weavejs_ops
+from weave.legacy.tests.util import geom, weavejs_ops
 
 
 def test_unnest():

diff --git a/weave/legacy/weave/ops_primitives/test_typeddict.py b/weave/legacy/weave/ops_primitives/test_typeddict.py
@@ -5,7 +5,7 @@
 from weave.legacy.weave import box
 from weave.legacy.weave.arrow.arrow import ArrowWeaveListType
 from weave.legacy.weave.language_features.tagging import tag_store, tagged_value_type
-from weave.legacy.weave.tests.util.list_arrow_test_helpers import ArrowNode
+from weave.legacy.tests.util.list_arrow_test_helpers import ArrowNode
 
 pick_options = [
     # Basic Pick

diff --git a/weave/tests/fixture_fakewandb.py b/weave/tests/fixture_fakewandb.py
@@ -11,6 +11,7 @@
 import wandb
 
 import weave
+from weave.legacy.tests.util.tag_test_util import op_add_tag
 
 # Note: We're mocking out the whole io_service right now. This is too
 # high level and doesn't test the actual io implementation. We should
@@ -22,7 +23,6 @@
     WeaveWBArtifactByIDURI,
     WeaveWBArtifactURI,
 )
-from weave.legacy.weave.tests.util.tag_test_util import op_add_tag
 
 TEST_TABLE_ARTIFACT_PATH = "testdata/wb_artifacits/test_res_1fwmcd3q:v0"
 ABS_TEST_TABLE_ARTIFACT_PATH = os.path.abspath(TEST_TABLE_ARTIFACT_PATH)