Fixed minor issues with tests

1 files changed, 2 insertions, 3 deletions

diff --git a/tests/test_schedulers.py b/tests/test_schedulers.py
index e6f454a..e3e9d26 100644
--- a/tests/test_schedulers.py
+++ b/tests/test_schedulers.py
@@ -207,7 +207,7 @@ class TestBaseScheduler(object):
         assert 'default' in scheduler._executors
         assert 'default' in scheduler._jobstores
 
-        scheduler._real_add_job.assert_called_once(job, 'store1', False)
+        scheduler._real_add_job.assert_called_once_with(job, 'store1', False, False)
         assert scheduler._pending_jobs == []
 
         assert scheduler._dispatch_event.call_count == 3
@@ -265,7 +265,6 @@ class TestBaseScheduler(object):
         else:
             assert executor.start.call_count == 0
 
-
     def test_add_executor_already_exists(self, scheduler):
         executor = DebugExecutor()
         scheduler.add_executor(executor)
@@ -381,7 +380,7 @@ class TestBaseScheduler(object):
 
         job._modify.assert_called_once_with(misfire_grace_time=5, max_instances=2, next_run_time=datetime(2014, 10, 17))
         if not pending:
-            jobstore.update_job.assert_called_once(job)
+            jobstore.update_job.assert_called_once_with(job)
 
         assert scheduler._dispatch_event.call_count == 1
         event = scheduler._dispatch_event.call_args[0][0]