test.unit.test_model_optimizer

Complexity

Total Complexity

2

Size/Duplication

Total Lines	24
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TestBuildOptimizer.test_build_optimizer_adam()	0	5	1
A	TestBuildOptimizer.test_build_optimizer_sgd()	0	5	1

# coding=utf-8

"""
Tests for deepreg/model/optimizer.py
pytest style
"""
import tensorflow as tf

import deepreg.model.optimizer as optimizer


class TestBuildOptimizer:
    def test_build_optimizer_adam(self):
        """Build an Adam optimizer"""
        opt_config = {"name": "Adam", "learning_rate": 1.0e-5}
        opt_get = optimizer.build_optimizer(opt_config)
        assert isinstance(opt_get, tf.keras.optimizers.Adam)

    def test_build_optimizer_sgd(self):
        """Build an SGD optimizer"""
        opt_config = {"name": "SGD"}
        opt_get = optimizer.build_optimizer(opt_config)
        assert isinstance(opt_get, tf.keras.optimizers.SGD)