From 6386072d8cbe288751b1e68492c3c9ef8e10e003 Mon Sep 17 00:00:00 2001
From: Charles Stern <62192187+cisaacstern@users.noreply.github.com>
Date: Mon, 9 Sep 2024 20:41:44 -0700
Subject: [PATCH] move test graph to ns pkg

---
 .../core/ecoscope_workflows/core/graph.py              |  4 ++--
 .../ecoscope-workflows/core/tests}/test_graph.py       | 10 +++-------
 2 files changed, 5 insertions(+), 9 deletions(-)
 rename {tests => src/ecoscope-workflows/core/tests}/test_graph.py (95%)

diff --git a/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py b/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py
index cf6389c92..5ab81e0fb 100644
--- a/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py
+++ b/src/ecoscope-workflows/core/ecoscope_workflows/core/graph.py
@@ -3,8 +3,8 @@
 from graphlib import TopologicalSorter
 from typing import Any, Literal
 
-from ecoscope_workflows.decorators import AsyncTask
-from ecoscope_workflows.executors import Future, FutureSequence
+from ecoscope_workflows.core.decorators import AsyncTask
+from ecoscope_workflows.core.executors import Future, FutureSequence
 
 logger = logging.getLogger(__name__)
 
diff --git a/tests/test_graph.py b/src/ecoscope-workflows/core/tests/test_graph.py
similarity index 95%
rename from tests/test_graph.py
rename to src/ecoscope-workflows/core/tests/test_graph.py
index f1aca5c67..fe2e89e94 100644
--- a/tests/test_graph.py
+++ b/src/ecoscope-workflows/core/tests/test_graph.py
@@ -1,14 +1,10 @@
 from dataclasses import dataclass
 from functools import lru_cache
 from typing import Sequence, TypeVar
-# from textwrap import dedent
 
-# import yaml
-
-# from ecoscope_workflows.compiler import Spec, TaskInstance
-from ecoscope_workflows.decorators import task
-from ecoscope_workflows.executors import Future, FutureSequence, LithopsExecutor
-from ecoscope_workflows.graph import DependsOn, DependsOnSequence, Graph, Node
+from ecoscope_workflows.core.decorators import task
+from ecoscope_workflows.core.executors import Future, FutureSequence, LithopsExecutor
+from ecoscope_workflows.core.graph import DependsOn, DependsOnSequence, Graph, Node
 
 T = TypeVar("T")