b3e1438a8271a10f7b6d0069a692c7b534d69c89,tensorflow_datasets/core/download/download_manager_test.py,DownloadManagerTest,test_extract_twice_parallel,#DownloadManagerTest#,183

Before Change

    if sys.version_info[0] > 2:
      self.assertTrue(res1 is res2)
    else:
      self.assertTrue(res1._promise is res2._promise)
    extracted_new.set()
    self.assertEqual(res1.get(), "/extract_dir/TAR.foo")
    self.assertEqual(1, self.extractor_extract.call_count)

After Change

        _get_promise_on_event("/extract_dir/TAR.foo"))
    manager = self._get_manager()
    extracted_new.set()
    out1 = manager.extract(["/dl_dir/foo.tar", "/dl_dir/foo.tar"])
    out2 = manager.extract("/dl_dir/foo.tar")
    self.assertEqual(out1[0], "/extract_dir/TAR.foo")
    self.assertEqual(out1[1], "/extract_dir/TAR.foo")
    self.assertEqual(out2, "/extract_dir/TAR.foo")

Instances