def test_archive_remote_data(self): runner = EMRJobRunner() tar_gz_path = runner._dir_archive_path('s3://walrus/archive') self.assertEqual(os.path.basename(tar_gz_path), 'archive.tar.gz') runner._create_dir_archive('s3://walrus/archive') tar_gz = tarfile.open(tar_gz_path, 'r:gz') try: self.assertEqual(sorted(tar_gz.getnames()), [os.path.join('bar', 'baz'), 'foo']) finally: tar_gz.close()