1#!/usr/bin/env python
2#
3# Copyright 2020, Data61, CSIRO (ABN 41 687 119 230)
4#
5# SPDX-License-Identifier: BSD-2-Clause
6#
7
8from __future__ import absolute_import, division, print_function, \
9    unicode_literals
10
11import capdl
12from tests import CapdlTestCase
13
14
15class TestSpec(CapdlTestCase):
16
17    def test_merge(self):
18        a = capdl.TCB('foo')
19        b = capdl.TCB('bar')
20
21        spec1 = capdl.Spec()
22        spec2 = capdl.Spec()
23        spec1.merge(spec2)
24        assert len(spec1.objs) == 0
25
26        spec1 = capdl.Spec()
27        spec1.add_object(a)
28        spec2 = capdl.Spec()
29        spec1.merge(spec2)
30        assert spec1.objs == set([a])
31
32        spec1 = capdl.Spec()
33        spec1.add_object(a)
34        spec2 = capdl.Spec()
35        spec2.add_object(b)
36        spec1.merge(spec2)
37        assert spec1.objs == set([a, b])
38