From 1d8486c7dc2cdace45142fec80e8c5f1a79610fc Mon Sep 17 00:00:00 2001
From: Jitendra Gundaniya <jitendra_gundaniya@mckinsey.com>
Date: Thu, 25 Apr 2024 19:11:02 +0100
Subject: [PATCH] Unit test fix

Signed-off-by: <>
---
 package/tests/test_api/test_rest/test_responses.py  |  2 +-
 .../test_repositories/test_modular_pipelines.py     | 13 ++++++++++---
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/package/tests/test_api/test_rest/test_responses.py b/package/tests/test_api/test_rest/test_responses.py
index ca2d65731d..1842ee51f1 100644
--- a/package/tests/test_api/test_rest/test_responses.py
+++ b/package/tests/test_api/test_rest/test_responses.py
@@ -206,7 +206,7 @@ def assert_example_data(response_data):
         {
             "id": "uk",
             "name": "uk",
-            "tags": ["split","train"],
+            "tags": ["split", "train"],
             "pipelines": ["__default__"],
             "type": "modularPipeline",
             "modular_pipelines": None,
diff --git a/package/tests/test_data_access/test_repositories/test_modular_pipelines.py b/package/tests/test_data_access/test_repositories/test_modular_pipelines.py
index f38d932524..ec9e722e04 100644
--- a/package/tests/test_data_access/test_repositories/test_modular_pipelines.py
+++ b/package/tests/test_data_access/test_repositories/test_modular_pipelines.py
@@ -4,7 +4,7 @@
 
 from kedro_viz.constants import ROOT_MODULAR_PIPELINE_ID
 from kedro_viz.data_access.repositories import ModularPipelinesRepository
-from kedro_viz.models.flowchart import GraphNode
+from kedro_viz.models.flowchart import GraphNode, GraphNodeType, ModularPipelineChild
 
 
 class TestModularPipelinesRepository:
@@ -42,15 +42,22 @@ def test_tags_inheritance(self, identity):
                 identity,
                 inputs="x",
                 outputs=None,
-                namespace="data_science",
+                namespace="parent",
                 tags={"tag1", "tag2"},
             )
         )
+
         modular_pipelines = ModularPipelinesRepository()
+        modular_pipelines.add_child(
+            "parent",
+            ModularPipelineChild(
+                id="parent.child", type=GraphNodeType.MODULAR_PIPELINE
+            ),
+        )
 
         modular_pipelines.extract_from_node(task_node)
         modular_pipeline = modular_pipelines.get_or_create_modular_pipeline(
-            "data_science"
+            "parent"
         )
 
         assert "tag1" in modular_pipeline.tags