aboutsummaryrefslogtreecommitdiff
path: root/test/doit/test_doit.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/doit/test_doit.py')
-rw-r--r--test/doit/test_doit.py25
1 files changed, 17 insertions, 8 deletions
diff --git a/test/doit/test_doit.py b/test/doit/test_doit.py
index 4076af8..8c98345 100644
--- a/test/doit/test_doit.py
+++ b/test/doit/test_doit.py
@@ -1,16 +1,18 @@
import inspect
-import logging
import os
import os.path
+import logging
+from pathlib import Path
+from ee.fact import DataSetManager
from inspect import Parameter
-
import ee.kicad.doit
-from ee.kicad.doit import task_kicad_sch_to_data_set, task_kicad_pcb_to_data_set, task_kicad_create_component_data_set
+
+from ee.kicad.doit import *
logger = logging.getLogger(__name__)
-file_dir = os.path.dirname(os.path.abspath(__file__))
-schematics_dir = os.path.join(file_dir, "schematics")
+file_dir = Path(__file__).parent
+schematics_dir = file_dir / "schematics"
def exec_task(task):
@@ -31,9 +33,16 @@ def exec_task(task):
# noinspection SpellCheckingInspection
def test_doit(tmpdir, caplog):
- ee.kicad.doit.init(sch=os.path.join(schematics_dir, "schematic-1.sch"),
- kicad_pcb=os.path.join(schematics_dir, "schematic-1.kicad_pcb"),
- data_set_dir=os.path.join(tmpdir, "ee"), )
+ dsm = DataSetManager(os.path.join(tmpdir, "ee"))
+
+ dsm.add_ds("csv", "kicad-footprint", "kicad-footprint-mapping",
+ path=schematics_dir / "ee" / "kicad-footprint.csv")
+
+ ee.kicad.doit.change_data_sets_for_task(task_kicad_create_component_data_set, lambda ds: ds + ["kicad-footprint"])
+
+ ee.kicad.doit.init(data_set_manager=dsm,
+ sch=schematics_dir / "schematic-1.sch",
+ kicad_pcb=schematics_dir / "schematic-1.kicad_pcb",)
exec_task(task_kicad_sch_to_data_set())
exec_task(task_kicad_pcb_to_data_set())