From 70dbc09a8c191ba5d6bb7745849ef2de13990109 Mon Sep 17 00:00:00 2001
From: jrybicki-jsc <j.rybicki@fz-juelich.de>
Date: Wed, 1 Sep 2021 11:45:34 +0200
Subject: [PATCH] fixing tests
---
tests/test_b2shareoperator.py | 3 ++-
tests/test_dag.py | 2 +-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/test_b2shareoperator.py b/tests/test_b2shareoperator.py
index 118b8af..e5de98e 100644
--- a/tests/test_b2shareoperator.py
+++ b/tests/test_b2shareoperator.py
@@ -9,10 +9,11 @@ TEST_DAG_ID = 'test_my_custom_operator'
class B2ShareOperatorTest(unittest.TestCase):
def setUp(self):
- self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily',name='testing_dag', default_args={'start_date' : DEFAULT_DATE})
+ self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : DEFAULT_DATE})
self.op = B2ShareOperator(
dag=self.dag,
task_id='test',
+ name='test_name'
)
self.ti = TaskInstance(task=self.op, execution_date=DEFAULT_DATE)
diff --git a/tests/test_dag.py b/tests/test_dag.py
index 08c39ad..8a1faa1 100644
--- a/tests/test_dag.py
+++ b/tests/test_dag.py
@@ -10,4 +10,4 @@ class TestADag(unittest.TestCase):
dag = self.dagbag.get_dag(dag_id='firsto')
assert self.dagbag.import_errors == {}
assert dag is not None
- assert len(dag.tasks) == 4
\ No newline at end of file
+ self.assertEqual(len(dag.tasks), 3, f"Actually: {len(dag.tasks)}")
\ No newline at end of file
--
GitLab