Source code for kitcar_ml.onboarding.test.test_onboarding_net

import torch

from kitcar_ml.onboarding.model import OnboardingNet


[docs]def test_with_dummy_input():
    print("Test Onboarding Net with dummy input")

    model = OnboardingNet()

    dummy_input = torch.randn([1, 1, 28, 28])
    output = model(dummy_input)

    assert output.size() == torch.Size(
        [1, 10]
    ), f"Output has wrong dimension: {output.size()}"


[docs]def main():
    print("Test Onboarding Net")
    test_with_dummy_input()


if __name__ == "__main__":
    main()