()
| 52 | |
| 53 | |
| 54 | def test_gat(): |
| 55 | script = os.path.join(EXAMPLE_ROOT, "gat.py") |
| 56 | out = subprocess.run(["python", str(script)], capture_output=True) |
| 57 | assert ( |
| 58 | out.returncode == 0 |
| 59 | ), f"stdout: {out.stdout.decode('utf-8')}\nstderr: {out.stderr.decode('utf-8')}" |
| 60 | stdout = out.stdout.decode("utf-8") |
| 61 | assert float(stdout[-5:]) > 0.7 |
| 62 | |
| 63 | |
| 64 | def test_hgnn(): |