Digital Drosophila Archive
39 lines
2.0 KiB
Python
39 lines
2.0 KiB
Python
"""Additional adversarial operator tests and durable unit-test record."""
|
|
import io
|
|
import json
|
|
from pathlib import Path
|
|
import sys
|
|
import unittest
|
|
import numpy as np
|
|
|
|
MODEL = Path('/data/digital-drosophila/models/male-cns-lif/terminal-scope-diagnostic-v1')
|
|
EXP = Path('/data/digital-drosophila/experiments/phase4ab-terminal-scope')
|
|
sys.path.insert(0, str(MODEL))
|
|
from scope import blocked_weights
|
|
import diagnostic
|
|
|
|
_, _, _, _, save, digest, _, _ = diagnostic.load_context()
|
|
stream = io.StringIO()
|
|
suite = unittest.defaultTestLoader.discover(str(MODEL), pattern='test_scope.py')
|
|
result = unittest.TextTestRunner(stream=stream, verbosity=2).run(suite)
|
|
assert result.wasSuccessful(), stream.getvalue()
|
|
rng = np.random.default_rng(41337)
|
|
raw = rng.integers(1, 100000, size=10000, dtype=np.int64)
|
|
contacts = np.asarray([rng.integers(0, n+1) for n in raw], dtype=np.int64)
|
|
base = (raw.astype(np.float32)*np.float32(.275)*rng.choice([-1, 0, 1], size=len(raw))).astype(np.float32)
|
|
raw_bytes, base_bytes = raw.tobytes(), base.tobytes()
|
|
out = blocked_weights(base, raw, contacts)
|
|
assert np.isfinite(out).all() and np.all(np.abs(out) <= np.abs(base))
|
|
assert np.all(out[contacts == raw] == 0)
|
|
assert out[contacts == 0].tobytes() == base[contacts == 0].tobytes()
|
|
assert np.all(np.sign(out[out != 0]) == np.sign(base[out != 0]))
|
|
assert raw_bytes == raw.tobytes() and base_bytes == base.tobytes()
|
|
for bad in [np.array([np.inf], np.float32), np.array([np.nan], np.float32)]:
|
|
try: blocked_weights(bad, np.array([2], np.int64), np.array([1], np.int64))
|
|
except ValueError: pass
|
|
else: raise AssertionError('nonfinite accepted')
|
|
save(EXP/'regression/unit-tests.json', {'pass': True, 'unit_tests': result.testsRun, 'fuzz_cases': len(raw),
|
|
'fuzz_seed': 41337, 'finite': True, 'never_sign_flip_or_amplify': True, 'input_nonmutation': True,
|
|
'zero_identity_bitwise': True, 'unittest_log': stream.getvalue(), 'source_sha256': digest(Path(__file__))})
|
|
print('PASS', result.testsRun, 'unit tests +', len(raw), 'randomized operator cases')
|