|
from tensorboardX import SummaryWriter |
|
import unittest |
|
from tensorboardX.record_writer import S3RecordWriter, make_valid_tf_name |
|
import os |
|
import boto3 |
|
from moto import mock_s3 |
|
|
|
os.environ.setdefault("AWS_ACCESS_KEY_ID", "foobar_key") |
|
os.environ.setdefault("AWS_SECRET_ACCESS_KEY", "foobar_secret") |
|
|
|
|
|
class RecordWriterTest(unittest.TestCase): |
|
@mock_s3 |
|
def test_record_writer_s3(self): |
|
client = boto3.client('s3', region_name='us-east-1') |
|
client.create_bucket(Bucket='this') |
|
writer = S3RecordWriter('s3://this/is/apen') |
|
bucket, path = writer.bucket_and_path() |
|
assert bucket == 'this' |
|
assert path == 'is/apen' |
|
writer.write(bytes(42)) |
|
writer.flush() |
|
|
|
def test_make_valid_tf_name(self): |
|
newname = make_valid_tf_name('$ave/&sound') |
|
assert newname == '._ave/_sound' |
|
|
