Permalink
Browse files

added ffmpeg to dockerfile and fixed test (#325)

* added ffmpeg to dockerfile and fixed test

* use ubuntu media repo for ffmpeg

* ffmpeg on travis 🍰

* install libav instead
  • Loading branch information...
kashif authored and jonasschneider committed Aug 29, 2016
1 parent a84edbc commit ebb2f2927388d90d266a13b0d713fe3765286fcc
Showing with 12 additions and 11 deletions.
  1. +11 −11 gym/monitoring/tests/test_video_recorder.py
  2. +1 −0 test.dockerfile
@@ -21,17 +21,17 @@ class UnrecordableEnv(object):
def render(self, mode=None):
pass
# TODO(jonas): disabled until we have ffmpeg on travis
# def test_record_simple():
# rec = VideoRecorder()
# env, id = gym.make("CartPole")
# rec.capture_frame(env)
# rec.close()
# assert not rec.empty
# assert not rec.broken
# assert os.path.exists(rec.path)
# f = open(rec.path)
# assert os.fstat(f.fileno()).st_size > 100
def test_record_simple():
env = gym.make("CartPole-v1")
rec = VideoRecorder(env)
env.reset()
rec.capture_frame()
rec.close()
assert not rec.empty
assert not rec.broken
assert os.path.exists(rec.path)
f = open(rec.path)
assert os.fstat(f.fileno()).st_size > 100
def test_no_frames():
env = BrokenRecordableEnv()
View
@@ -19,6 +19,7 @@ RUN apt-get update \
unzip \
git \
xpra \
libav-tools \
python3-dev \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* \

0 comments on commit ebb2f29

Please sign in to comment.