init public release

tests/unit/test_generate_import_script.py

@@ -0,0 +1,78 @@
+"""Unit tests for ``generate_neo4j_import_script``.
+
+It scans a directory of BioCypher CSV output, classifies each ``*-header.csv``
+as a node or relationship, and emits a ``neo4j-admin import`` shell script.
+We drive it with a temporary directory so no real ``/neo4j_import`` is needed.
+"""
+
+import os
+import stat
+
+import import_fhir_to_nx_diGraph as pipeline
+
+
+def _touch(directory, name):
+    (directory / name).write_text("")
+
+
+def _generate(tmp_path):
+    ret = pipeline.generate_neo4j_import_script(
+        directory_path=str(tmp_path) + os.sep,
+        output_file="neo4j-admin-import-call.sh",
+    )
+    out = tmp_path / "neo4j-admin-import-call.sh"
+    return out.read_text(), out, ret
+
+
+def test_node_with_parts_is_included_as_nodes(tmp_path):
+    _touch(tmp_path, "Patient-header.csv")
+    _touch(tmp_path, "Patient-part0.csv")
+    script, _, _ = _generate(tmp_path)
+    assert '--nodes="/neo4j_import/Patient-header.csv,/neo4j_import/Patient-part.*"' in script
+
+
+def test_association_is_classified_as_relationship(tmp_path):
+    _touch(tmp_path, "PatientToObservationAssociation-header.csv")
+    _touch(tmp_path, "PatientToObservationAssociation-part0.csv")
+    script, _, _ = _generate(tmp_path)
+    assert "--relationships=" in script
+    assert "PatientToObservationAssociation" in script
+
+
+def test_has_pattern_is_classified_as_relationship(tmp_path):
+    _touch(tmp_path, "Encounter_has_Diagnosis-header.csv")
+    _touch(tmp_path, "Encounter_has_Diagnosis-part0.csv")
+    script, _, _ = _generate(tmp_path)
+    assert "--relationships=" in script
+    assert "--nodes=" not in script
+
+
+def test_header_without_part_files_is_dropped(tmp_path):
+    # SHARP EDGE: entities are only emitted when a matching *-part* file
+    # exists. A lone header silently produces no import command. Pinning so
+    # the behaviour is intentional, not a surprise empty graph.
+    _touch(tmp_path, "Patient-header.csv")  # no part file
+    script, _, _ = _generate(tmp_path)
+    assert "--nodes=" not in script
+    assert "--relationships=" not in script
+
+
+def test_script_is_written_and_executable(tmp_path):
+    _touch(tmp_path, "Patient-header.csv")
+    _touch(tmp_path, "Patient-part0.csv")
+    _, out, ret = _generate(tmp_path)
+    assert out.exists()
+    mode = os.stat(out).st_mode
+    assert mode & stat.S_IXUSR  # owner-executable
+    # SHARP EDGE: the docstring promises it returns the script path, but the
+    # function has no `return`, so callers actually get None. Pinning the
+    # real behaviour; fix the function (or the docstring) and flip this.
+    assert ret is None
+
+
+def test_script_contains_version_branching(tmp_path):
+    _touch(tmp_path, "Patient-header.csv")
+    _touch(tmp_path, "Patient-part0.csv")
+    script, _, _ = _generate(tmp_path)
+    assert "neo4j-admin database import full" in script  # v5+ branch
+    assert "neo4j-admin import --database=neo4j" in script  # legacy branch