fix print call count assertions

Author: dlawin

tests/test_dbt.py

@@ -499,7 +499,7 @@ def test_cloud_diff(self, mock_api, mock_os_environ, mock_print):
         _cloud_diff(diff_vars, expected_datasource_id, api=mock_api)
 
         mock_api.create_data_diff.assert_called_once()
-        mock_print.assert_called_once()
+        self.assertEqual(mock_print.call_count, 2)
 
         payload = mock_api.create_data_diff.call_args[1]["payload"]
         self.assertEqual(payload.data_source1_id, expected_datasource_id)
@@ -541,7 +541,7 @@ def test_diff_is_cloud(
         mock_initialize_api.assert_called_once()
         mock_cloud_diff.assert_called_once_with(expected_diff_vars, 1, api)
         mock_local_diff.assert_not_called()
-        mock_print.assert_not_called()
+        mock_print.assert_called_once()
 
     @patch("data_diff.dbt._initialize_api")
     @patch("data_diff.dbt._get_diff_vars")
@@ -722,7 +722,7 @@ def test_diff_is_cloud_no_pks(
         mock_dbt_parser_inst.set_connection.assert_not_called()
         mock_cloud_diff.assert_not_called()
         mock_local_diff.assert_not_called()
-        self.assertEqual(mock_print.call_count, 1)
+        self.assertEqual(mock_print.call_count, 2)
 
     @patch("data_diff.dbt._get_diff_vars")
     @patch("data_diff.dbt._local_diff")