|
|
|
|
|
|
import unittest.mock as mock |
|
|
|
import pytest |
|
|
|
import os |
|
|
|
brain_parameters, brain_infos, total_expected = load_demonstration('./tests/trainers/test.demo') |
|
|
|
path_prefix = os.path.dirname(os.path.abspath(__file__)) |
|
|
|
brain_parameters, brain_infos, total_expected = load_demonstration(path_prefix+'/test.demo') |
|
|
|
assert (brain_parameters.brain_name == "Ball3DBrain") |
|
|
|
assert (brain_parameters.vector_observation_space_size == 8) |
|
|
|
assert (len(brain_infos) == total_expected) |
|
|
|