xref: /aosp_15_r20/external/google-fruit/tests/meta/test_meta_component.py (revision a65addddcf69f38db5b288d787b6b7571a57bb8f) 
1#!/usr/bin/env python3
2#  Copyright 2016 Google Inc. All Rights Reserved.
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#      http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS-IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15
16from absl.testing import parameterized
17from fruit_test_common import *
18
19COMMON_DEFINITIONS = '''
20    #define IN_FRUIT_CPP_FILE 1
21
22    #include "meta/common.h"
23    #include <fruit/impl/meta/component.h>
24
25    struct A1 {};
26    struct B1 {};
27
28    using A = Type<A1>;
29    using B = Type<B1>;
30
31    using AssistedA = Type<Assisted<A1>>;
32    using AssistedB = Type<Assisted<B1>>;
33    '''
34
35class TestMetaComponent(parameterized.TestCase):
36    def test_NumAssisted(self):
37        source = '''
38            int main() {
39              AssertSame(Int<0>, NumAssisted(Vector<>));
40              AssertSame(Int<0>, NumAssisted(Vector<A>));
41              AssertSame(Int<1>, NumAssisted(Vector<AssistedA>));
42              AssertSame(Int<0>, NumAssisted(Vector<A, B>));
43              AssertSame(Int<1>, NumAssisted(Vector<AssistedA, B>));
44              AssertSame(Int<1>, NumAssisted(Vector<A, AssistedB>));
45              AssertSame(Int<2>, NumAssisted(Vector<AssistedA, AssistedB>));
46            }
47            '''
48        expect_success(
49            COMMON_DEFINITIONS,
50            source,
51            locals())
52
53    def test_NumAssistedBefore(self):
54        source = '''
55            int main() {
56              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<>));
57
58              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<A>));
59              AssertSame(Int<0>, NumAssistedBefore(Int<1>, Vector<A>));
60
61              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<AssistedA>));
62              AssertSame(Int<1>, NumAssistedBefore(Int<1>, Vector<AssistedA>));
63
64              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<A, B>));
65              AssertSame(Int<0>, NumAssistedBefore(Int<1>, Vector<A, B>));
66              AssertSame(Int<0>, NumAssistedBefore(Int<2>, Vector<A, B>));
67
68              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<AssistedA, B>));
69              AssertSame(Int<1>, NumAssistedBefore(Int<1>, Vector<AssistedA, B>));
70              AssertSame(Int<1>, NumAssistedBefore(Int<2>, Vector<AssistedA, B>));
71
72              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<A, AssistedB>));
73              AssertSame(Int<0>, NumAssistedBefore(Int<1>, Vector<A, AssistedB>));
74              AssertSame(Int<1>, NumAssistedBefore(Int<2>, Vector<A, AssistedB>));
75
76              AssertSame(Int<0>, NumAssistedBefore(Int<0>, Vector<AssistedA, AssistedB>));
77              AssertSame(Int<1>, NumAssistedBefore(Int<1>, Vector<AssistedA, AssistedB>));
78              AssertSame(Int<2>, NumAssistedBefore(Int<2>, Vector<AssistedA, AssistedB>));
79            }
80            '''
81        expect_success(
82            COMMON_DEFINITIONS,
83            source,
84            locals())
85
86if __name__ == '__main__':
87    absltest.main()
88