def METHOD_NAME():
    """Quick test of AtomicConv."""
    acm = atomic_conv.AtomicConvModel(n_tasks=1,
                                      batch_size=1,
                                      layer_sizes=[
                                          1,
                                      ],
                                      frag1_num_atoms=5,
                                      frag2_num_atoms=5,
                                      complex_num_atoms=10)
    assert acm.complex_num_atoms == 10
    assert len(acm.atom_types) == 15