Add test_components
This commit is contained in:
parent
ee30d4da5b
commit
1e23ba05fa
25
tests/test_components.py
Normal file
25
tests/test_components.py
Normal file
@ -0,0 +1,25 @@
|
||||
"""ProtoTorch components test suite."""
|
||||
|
||||
import prototorch as pt
|
||||
import torch
|
||||
|
||||
|
||||
def test_labcomps_zeros_init():
|
||||
protos = torch.zeros(3, 2)
|
||||
c = pt.components.LabeledComponents(
|
||||
distribution=[1, 1, 1],
|
||||
initializer=pt.components.Zeros(2),
|
||||
)
|
||||
assert (c.components == protos).any() == True
|
||||
|
||||
|
||||
def test_labcomps_warmstart():
|
||||
protos = torch.randn(3, 2)
|
||||
plabels = torch.tensor([1, 2, 3])
|
||||
c = pt.components.LabeledComponents(
|
||||
distribution=[1, 1, 1],
|
||||
initializer=None,
|
||||
initialized_components=[protos, plabels],
|
||||
)
|
||||
assert (c.components == protos).any() == True
|
||||
assert (c.component_labels == plabels).any() == True
|
Loading…
Reference in New Issue
Block a user