Unity でUnitTest とML-Agents を使用する時の参照トラブル対応


Unity で単体テスト環境を構築した後、ML-Agents を使ってゲーム上のNPU をAI化しようとして嵌ったので、顛末を共有します。

概要

  • Unity で単体テスト環境を構築
  • ML-Agent をプロジェクトに追加してAgent クラスを参照しようとしたらnamespace が見つからない問題が発生
  • ML-Agent 内に鎮座していたAssembly Definision Filesを参照させるようにして問題解決

Unity で単体テスト環境を構築

テストコードを格納するためのディレクトリを作成する

まず、Unity のProject ツリーにて、テスト環境を格納するためのディレクトリを作成します。この時、通常のディレクトリ作成手順(Create -> Folder)ではなく、Create -> Testing -> Tests Assembly Folder と選択することでテスト環境用のディレクトリを作成します。

ディレクトリが作成された際、ディレクトリの中にTests.asmdef という名前のAssembly Definition Files (以下adf)が作られています。このadf によって、Unity のTestRunner がリンクされ、単体テストを実行できるようになります。適宜本ディレクトリ内にテストコードを追加していきましょう(本稿では単体テストそのものについては省略します)。

テスト対象のディレクトリを準備する

上記の手順でadf を作成すると、テストシナリオ用のdll (以下、テストシナリオ.dll)が作られるようになります。ここで、テストシナリオ.dll と周辺のdll との関係は以下の図のようになっています。

この時点では、テスト対象のコード群は図の「その他大勢」に含まれてしまっており、テストシナリオ.dllから参照することができません。そのため、テスト対象のコード群を独立したdll として分離し(以下、テスト対象.dll)、テストシナリオ.dll から参照するように設定する必要があります。下記がそのイメージ図です。

テスト対象を分離させるには、テスト対象のソースコードを特定のディレクトリ以下に配置し、このディレクトリ直下に新たにadf を作成します。今回はテスト対象のadf ファイル名をTarget.asmdef とします。

続いて、テストシナリオ側(Tests.asmdef)からテスト対象(Target.asmdef)を参照できるように、Unity のエディタ上でTarget.asmdef をTests.asmdef のAssembly Definition References に追加します。既にTestRunner がリストに存在していると思うので、リストに追加する形になります。

ML-Agent をプロジェクトに追加

GitHub のUnity-Technologies / ml-agentsを参考にして、ML-Agents をUnity のプロジェクトに追加してください。

そして、自分のソースコードからAgent クラスを参照しようとしたらnamespace が見つからないと怒られます。。。

using UnityEngine;
using UnityTetris.Abstract;
using UnityTetris.Interface;
using Unity.MLAgents;
using Unity.MLAgents.Sensors;

public class MLInput : Agent, IInputManager
{

「namespace ML-Agents が見つからない」と怒られますね。。。

今回の単体テスト環境構築により、自分のソースコードがTarget.asmdef によりAssembly-Csharp.dll (いわゆるその他大勢)から分離されてしまうため、ML-Agents を参照するように明示的に設定を行う必要があります。

ML-Agents を参照するように設定する

実は、プロジェクトビューのPackages 以下にあるML Agents 内にもUnity.ML-Agents.asmdef というadf があるので、これをTarget.asmdef から参照するように設定する必要があります。Target.asmdef をInspector で表示し、この中にあるAssembly Definition References のリストにUnity.ML-Agents.asmdef を追加してください。

これにより、自分のソースコードからML-Agents を参照することができるようになります。各dll の関係は以下のようになります。

ML-Agents 以外のパッケージを使用する場合でも同様の理屈で参照設定する必要がありますので、単体テスト環境を構築した場合は注意しましょう。

参考ページ