From 43cd89c872a5a02d042b8c7bd487cc74f443ca4a Mon Sep 17 00:00:00 2001
From: Sijia Wang <wangsijia0628@gmail.com>
Date: Mon, 7 Nov 2022 15:21:34 -0500
Subject: [PATCH] set up testing infrastructure (#40)

Co-authored-by: Lisa Zorn <lzorn@bayareametro.gov>
---
 test/auto_ownership/.gitkeep               |  0
 test/cdap/.gitkeep                         |  0
 test/conftest.py                           | 99 ++++++++++++++++++++++
 test/joint_tour_frequency/.gitkeep         |  0
 test/non_mandatory_tour_frequency/.gitkeep |  0
 test/parking_location/.gitkeep             |  0
 6 files changed, 99 insertions(+)
 create mode 100644 test/auto_ownership/.gitkeep
 create mode 100644 test/cdap/.gitkeep
 create mode 100644 test/conftest.py
 create mode 100644 test/joint_tour_frequency/.gitkeep
 create mode 100644 test/non_mandatory_tour_frequency/.gitkeep
 create mode 100644 test/parking_location/.gitkeep

diff --git a/test/auto_ownership/.gitkeep b/test/auto_ownership/.gitkeep
new file mode 100644
index 000000000..e69de29bb
diff --git a/test/cdap/.gitkeep b/test/cdap/.gitkeep
new file mode 100644
index 000000000..e69de29bb
diff --git a/test/conftest.py b/test/conftest.py
new file mode 100644
index 000000000..d69a6a83c
--- /dev/null
+++ b/test/conftest.py
@@ -0,0 +1,99 @@
+import os
+
+import orca
+import pandas as pd
+import pytest
+from activitysim.core import pipeline
+from activitysim.core.los import Network_LOS as los
+
+
+@pytest.fixture(scope='module')
+def initialize_pipeline(module: str, tables: dict[str, str], initialize_network_los: bool) -> pipeline.Pipeline:
+    test_dir = os.path.join('test', module)
+    configs_dir = os.path.join(test_dir, 'configs')
+    data_dir = os.path.join(test_dir, 'data')
+    output_dir = os.path.join(test_dir, 'output')
+
+    if os.path.isdir(configs_dir):
+        orca.add_injectable('configs_dir', configs_dir)
+
+    if os.path.isdir(data_dir):
+        orca.add_injectable('data_dir', data_dir)
+
+    if os.path.isdir(test_dir):
+        if not os.path.isdir(output_dir):
+            os.mkdir(output_dir)
+        orca.add_injectable('output_dir', output_dir)
+
+    # Read in the input test dataframes
+    for dataframe_name, idx_name in tables.items():
+        df = pd.read_csv(
+            os.path.join('test', module, 'data', f'{dataframe_name}.csv'), index_col=idx_name
+        )
+        orca.add_table(dataframe_name, df)
+
+    if initialize_network_los:
+        net_los = los()
+        net_los.load_data()
+        orca.add_injectable('network_los', net_los)
+
+    # Add an output directory in current working directory if it's not already there
+    try:
+        os.makedirs('output')
+    except FileExistsError:
+        # directory already exists
+        pass
+
+    # Add the dataframes to the pipeline
+    pipeline.open_pipeline()
+    pipeline.add_checkpoint(module)
+    pipeline.close_pipeline()
+
+    # By convention, this method needs to yield something
+    yield pipeline._PIPELINE
+
+    if pipeline.is_open():
+        pipeline.close_pipeline()
+
+@pytest.fixture(scope='module')
+def reconnect_pipeline(module: str, initialize_network_los: bool, load_checkpoint: str) -> pipeline.Pipeline:
+    test_dir = os.path.join('test', module)
+    configs_dir = os.path.join(test_dir, 'configs')
+    data_dir = os.path.join(test_dir, 'data')
+    output_dir = os.path.join(test_dir, 'output')
+
+    if os.path.isdir(configs_dir):
+        orca.add_injectable('configs_dir', configs_dir)
+
+    if os.path.isdir(data_dir):
+        orca.add_injectable('data_dir', data_dir)
+
+    if os.path.isdir(test_dir):
+        if not os.path.isdir(output_dir):
+            os.mkdir(output_dir)
+        orca.add_injectable('output_dir', output_dir)
+
+    # Read in the existing pipeline
+    orca.add_injectable('pipeline_file_path', output_dir)
+
+    if initialize_network_los:
+        net_los = los()
+        net_los.load_data()
+        orca.add_injectable('network_los', net_los)
+
+    # Add an output directory in current working directory if it's not already there
+    try:
+        os.makedirs('output')
+    except FileExistsError:
+        # directory already exists
+        pass
+
+    pipeline.open_pipeline(resume_after=load_checkpoint)
+    pipeline.add_checkpoint(module)
+    pipeline.close_pipeline()
+    # By convention, this method needs to yield something
+    yield pipeline._PIPELINE
+
+    # pytest teardown code
+    if pipeline.is_open():
+        pipeline.close_pipeline()
diff --git a/test/joint_tour_frequency/.gitkeep b/test/joint_tour_frequency/.gitkeep
new file mode 100644
index 000000000..e69de29bb
diff --git a/test/non_mandatory_tour_frequency/.gitkeep b/test/non_mandatory_tour_frequency/.gitkeep
new file mode 100644
index 000000000..e69de29bb
diff --git a/test/parking_location/.gitkeep b/test/parking_location/.gitkeep
new file mode 100644
index 000000000..e69de29bb