fixing tests

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)
 

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