1#!/usr/bin/env python 2# 3# Copyright 2006, Google Inc. 4# All rights reserved. 5# 6# Redistribution and use in source and binary forms, with or without 7# modification, are permitted provided that the following conditions are 8# met: 9# 10# * Redistributions of source code must retain the above copyright 11# notice, this list of conditions and the following disclaimer. 12# * Redistributions in binary form must reproduce the above 13# copyright notice, this list of conditions and the following disclaimer 14# in the documentation and/or other materials provided with the 15# distribution. 16# * Neither the name of Google Inc. nor the names of its 17# contributors may be used to endorse or promote products derived from 18# this software without specific prior written permission. 19# 20# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 23# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 24# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 25# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 26# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 30# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31 32"""gen_gtest_pred_impl.py v0.1 33 34Generates the implementation of Google Test predicate assertions and 35accompanying tests. 36 37Usage: 38 39 gen_gtest_pred_impl.py MAX_ARITY 40 41where MAX_ARITY is a positive integer. 42 43The command generates the implementation of up-to MAX_ARITY-ary 44predicate assertions, and writes it to file gtest_pred_impl.h in the 45directory where the script is. It also generates the accompanying 46unit test in file gtest_pred_impl_unittest.cc. 47""" 48 49__author__ = 'wan@google.com (Zhanyong Wan)' 50 51import os 52import sys 53import time 54 55# Where this script is. 56SCRIPT_DIR = os.path.dirname(sys.argv[0]) 57 58# Where to store the generated header. 59HEADER = os.path.join(SCRIPT_DIR, '../include/gtest/gtest_pred_impl.h') 60 61# Where to store the generated unit test. 62UNIT_TEST = os.path.join(SCRIPT_DIR, '../test/gtest_pred_impl_unittest.cc') 63 64 65def HeaderPreamble(n): 66 """Returns the preamble for the header file. 67 68 Args: 69 n: the maximum arity of the predicate macros to be generated. 70 """ 71 72 # A map that defines the values used in the preamble template. 73 DEFS = { 74 'today' : time.strftime('%m/%d/%Y'), 75 'year' : time.strftime('%Y'), 76 'command' : '%s %s' % (os.path.basename(sys.argv[0]), n), 77 'n' : n 78 } 79 80 return ( 81"""// Copyright 2006, Google Inc. 82// All rights reserved. 83// 84// Redistribution and use in source and binary forms, with or without 85// modification, are permitted provided that the following conditions are 86// met: 87// 88// * Redistributions of source code must retain the above copyright 89// notice, this list of conditions and the following disclaimer. 90// * Redistributions in binary form must reproduce the above 91// copyright notice, this list of conditions and the following disclaimer 92// in the documentation and/or other materials provided with the 93// distribution. 94// * Neither the name of Google Inc. nor the names of its 95// contributors may be used to endorse or promote products derived from 96// this software without specific prior written permission. 97// 98// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 99// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 100// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 101// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 102// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 103// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 104// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 105// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 106// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 107// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 108// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 109 110// This file is AUTOMATICALLY GENERATED on %(today)s by command 111// '%(command)s'. DO NOT EDIT BY HAND! 112// 113// Implements a family of generic predicate assertion macros. 114 115#ifndef GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ 116#define GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ 117 118#include "gtest/gtest.h" 119 120namespace testing { 121 122// This header implements a family of generic predicate assertion 123// macros: 124// 125// ASSERT_PRED_FORMAT1(pred_format, v1) 126// ASSERT_PRED_FORMAT2(pred_format, v1, v2) 127// ... 128// 129// where pred_format is a function or functor that takes n (in the 130// case of ASSERT_PRED_FORMATn) values and their source expression 131// text, and returns a testing::AssertionResult. See the definition 132// of ASSERT_EQ in gtest.h for an example. 133// 134// If you don't care about formatting, you can use the more 135// restrictive version: 136// 137// ASSERT_PRED1(pred, v1) 138// ASSERT_PRED2(pred, v1, v2) 139// ... 140// 141// where pred is an n-ary function or functor that returns bool, 142// and the values v1, v2, ..., must support the << operator for 143// streaming to std::ostream. 144// 145// We also define the EXPECT_* variations. 146// 147// For now we only support predicates whose arity is at most %(n)s. 148// Please email googletestframework@googlegroups.com if you need 149// support for higher arities. 150 151// GTEST_ASSERT_ is the basic statement to which all of the assertions 152// in this file reduce. Don't use this in your code. 153 154#define GTEST_ASSERT_(expression, on_failure) \\ 155 GTEST_AMBIGUOUS_ELSE_BLOCKER_ \\ 156 if (const ::testing::AssertionResult gtest_ar = (expression)) \\ 157 ; \\ 158 else \\ 159 on_failure(gtest_ar.failure_message()) 160""" % DEFS) 161 162 163def Arity(n): 164 """Returns the English name of the given arity.""" 165 166 if n < 0: 167 return None 168 elif n <= 3: 169 return ['nullary', 'unary', 'binary', 'ternary'][n] 170 else: 171 return '%s-ary' % n 172 173 174def Title(word): 175 """Returns the given word in title case. The difference between 176 this and string's title() method is that Title('4-ary') is '4-ary' 177 while '4-ary'.title() is '4-Ary'.""" 178 179 return word[0].upper() + word[1:] 180 181 182def OneTo(n): 183 """Returns the list [1, 2, 3, ..., n].""" 184 185 return range(1, n + 1) 186 187 188def Iter(n, format, sep=''): 189 """Given a positive integer n, a format string that contains 0 or 190 more '%s' format specs, and optionally a separator string, returns 191 the join of n strings, each formatted with the format string on an 192 iterator ranged from 1 to n. 193 194 Example: 195 196 Iter(3, 'v%s', sep=', ') returns 'v1, v2, v3'. 197 """ 198 199 # How many '%s' specs are in format? 200 spec_count = len(format.split('%s')) - 1 201 return sep.join([format % (spec_count * (i,)) for i in OneTo(n)]) 202 203 204def ImplementationForArity(n): 205 """Returns the implementation of n-ary predicate assertions.""" 206 207 # A map the defines the values used in the implementation template. 208 DEFS = { 209 'n' : str(n), 210 'vs' : Iter(n, 'v%s', sep=', '), 211 'vts' : Iter(n, '#v%s', sep=', '), 212 'arity' : Arity(n), 213 'Arity' : Title(Arity(n)) 214 } 215 216 impl = """ 217 218// Helper function for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use 219// this in your code. 220template <typename Pred""" % DEFS 221 222 impl += Iter(n, """, 223 typename T%s""") 224 225 impl += """> 226AssertionResult AssertPred%(n)sHelper(const char* pred_text""" % DEFS 227 228 impl += Iter(n, """, 229 const char* e%s""") 230 231 impl += """, 232 Pred pred""" 233 234 impl += Iter(n, """, 235 const T%s& v%s""") 236 237 impl += """) { 238 if (pred(%(vs)s)) return AssertionSuccess(); 239 240""" % DEFS 241 242 impl += ' return AssertionFailure() << pred_text << "("' 243 244 impl += Iter(n, """ 245 << e%s""", sep=' << ", "') 246 247 impl += ' << ") evaluates to false, where"' 248 249 impl += Iter(n, """ 250 << "\\n" << e%s << " evaluates to " << v%s""") 251 252 impl += """; 253} 254 255// Internal macro for implementing {EXPECT|ASSERT}_PRED_FORMAT%(n)s. 256// Don't use this in your code. 257#define GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, on_failure)\\ 258 GTEST_ASSERT_(pred_format(%(vts)s, %(vs)s), \\ 259 on_failure) 260 261// Internal macro for implementing {EXPECT|ASSERT}_PRED%(n)s. Don't use 262// this in your code. 263#define GTEST_PRED%(n)s_(pred, %(vs)s, on_failure)\\ 264 GTEST_ASSERT_(::testing::AssertPred%(n)sHelper(#pred""" % DEFS 265 266 impl += Iter(n, """, \\ 267 #v%s""") 268 269 impl += """, \\ 270 pred""" 271 272 impl += Iter(n, """, \\ 273 v%s""") 274 275 impl += """), on_failure) 276 277// %(Arity)s predicate assertion macros. 278#define EXPECT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\ 279 GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, GTEST_NONFATAL_FAILURE_) 280#define EXPECT_PRED%(n)s(pred, %(vs)s) \\ 281 GTEST_PRED%(n)s_(pred, %(vs)s, GTEST_NONFATAL_FAILURE_) 282#define ASSERT_PRED_FORMAT%(n)s(pred_format, %(vs)s) \\ 283 GTEST_PRED_FORMAT%(n)s_(pred_format, %(vs)s, GTEST_FATAL_FAILURE_) 284#define ASSERT_PRED%(n)s(pred, %(vs)s) \\ 285 GTEST_PRED%(n)s_(pred, %(vs)s, GTEST_FATAL_FAILURE_) 286 287""" % DEFS 288 289 return impl 290 291 292def HeaderPostamble(): 293 """Returns the postamble for the header file.""" 294 295 return """ 296 297} // namespace testing 298 299#endif // GTEST_INCLUDE_GTEST_GTEST_PRED_IMPL_H_ 300""" 301 302 303def GenerateFile(path, content): 304 """Given a file path and a content string 305 overwrites it with the given content. 306 """ 307 print 'Updating file %s . . .' % path 308 f = file(path, 'w+') 309 print >>f, content, 310 f.close() 311 312 print 'File %s has been updated.' % path 313 314 315def GenerateHeader(n): 316 """Given the maximum arity n, updates the header file that implements 317 the predicate assertions. 318 """ 319 GenerateFile(HEADER, 320 HeaderPreamble(n) 321 + ''.join([ImplementationForArity(i) for i in OneTo(n)]) 322 + HeaderPostamble()) 323 324 325def UnitTestPreamble(): 326 """Returns the preamble for the unit test file.""" 327 328 # A map that defines the values used in the preamble template. 329 DEFS = { 330 'today' : time.strftime('%m/%d/%Y'), 331 'year' : time.strftime('%Y'), 332 'command' : '%s %s' % (os.path.basename(sys.argv[0]), sys.argv[1]), 333 } 334 335 return ( 336"""// Copyright 2006, Google Inc. 337// All rights reserved. 338// 339// Redistribution and use in source and binary forms, with or without 340// modification, are permitted provided that the following conditions are 341// met: 342// 343// * Redistributions of source code must retain the above copyright 344// notice, this list of conditions and the following disclaimer. 345// * Redistributions in binary form must reproduce the above 346// copyright notice, this list of conditions and the following disclaimer 347// in the documentation and/or other materials provided with the 348// distribution. 349// * Neither the name of Google Inc. nor the names of its 350// contributors may be used to endorse or promote products derived from 351// this software without specific prior written permission. 352// 353// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 354// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 355// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 356// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 357// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 358// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 359// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 360// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 361// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 362// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 363// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 364 365// This file is AUTOMATICALLY GENERATED on %(today)s by command 366// '%(command)s'. DO NOT EDIT BY HAND! 367 368// Regression test for gtest_pred_impl.h 369// 370// This file is generated by a script and quite long. If you intend to 371// learn how Google Test works by reading its unit tests, read 372// gtest_unittest.cc instead. 373// 374// This is intended as a regression test for the Google Test predicate 375// assertions. We compile it as part of the gtest_unittest target 376// only to keep the implementation tidy and compact, as it is quite 377// involved to set up the stage for testing Google Test using Google 378// Test itself. 379// 380// Currently, gtest_unittest takes ~11 seconds to run in the testing 381// daemon. In the future, if it grows too large and needs much more 382// time to finish, we should consider separating this file into a 383// stand-alone regression test. 384 385#include <iostream> 386 387#include "gtest/gtest.h" 388#include "gtest/gtest-spi.h" 389 390// A user-defined data type. 391struct Bool { 392 explicit Bool(int val) : value(val != 0) {} 393 394 bool operator>(int n) const { return value > Bool(n).value; } 395 396 Bool operator+(const Bool& rhs) const { return Bool(value + rhs.value); } 397 398 bool operator==(const Bool& rhs) const { return value == rhs.value; } 399 400 bool value; 401}; 402 403// Enables Bool to be used in assertions. 404std::ostream& operator<<(std::ostream& os, const Bool& x) { 405 return os << (x.value ? "true" : "false"); 406} 407 408""" % DEFS) 409 410 411def TestsForArity(n): 412 """Returns the tests for n-ary predicate assertions.""" 413 414 # A map that defines the values used in the template for the tests. 415 DEFS = { 416 'n' : n, 417 'es' : Iter(n, 'e%s', sep=', '), 418 'vs' : Iter(n, 'v%s', sep=', '), 419 'vts' : Iter(n, '#v%s', sep=', '), 420 'tvs' : Iter(n, 'T%s v%s', sep=', '), 421 'int_vs' : Iter(n, 'int v%s', sep=', '), 422 'Bool_vs' : Iter(n, 'Bool v%s', sep=', '), 423 'types' : Iter(n, 'typename T%s', sep=', '), 424 'v_sum' : Iter(n, 'v%s', sep=' + '), 425 'arity' : Arity(n), 426 'Arity' : Title(Arity(n)), 427 } 428 429 tests = ( 430"""// Sample functions/functors for testing %(arity)s predicate assertions. 431 432// A %(arity)s predicate function. 433template <%(types)s> 434bool PredFunction%(n)s(%(tvs)s) { 435 return %(v_sum)s > 0; 436} 437 438// The following two functions are needed to circumvent a bug in 439// gcc 2.95.3, which sometimes has problem with the above template 440// function. 441bool PredFunction%(n)sInt(%(int_vs)s) { 442 return %(v_sum)s > 0; 443} 444bool PredFunction%(n)sBool(%(Bool_vs)s) { 445 return %(v_sum)s > 0; 446} 447""" % DEFS) 448 449 tests += """ 450// A %(arity)s predicate functor. 451struct PredFunctor%(n)s { 452 template <%(types)s> 453 bool operator()(""" % DEFS 454 455 tests += Iter(n, 'const T%s& v%s', sep=""", 456 """) 457 458 tests += """) { 459 return %(v_sum)s > 0; 460 } 461}; 462""" % DEFS 463 464 tests += """ 465// A %(arity)s predicate-formatter function. 466template <%(types)s> 467testing::AssertionResult PredFormatFunction%(n)s(""" % DEFS 468 469 tests += Iter(n, 'const char* e%s', sep=""", 470 """) 471 472 tests += Iter(n, """, 473 const T%s& v%s""") 474 475 tests += """) { 476 if (PredFunction%(n)s(%(vs)s)) 477 return testing::AssertionSuccess(); 478 479 return testing::AssertionFailure() 480 << """ % DEFS 481 482 tests += Iter(n, 'e%s', sep=' << " + " << ') 483 484 tests += """ 485 << " is expected to be positive, but evaluates to " 486 << %(v_sum)s << "."; 487} 488""" % DEFS 489 490 tests += """ 491// A %(arity)s predicate-formatter functor. 492struct PredFormatFunctor%(n)s { 493 template <%(types)s> 494 testing::AssertionResult operator()(""" % DEFS 495 496 tests += Iter(n, 'const char* e%s', sep=""", 497 """) 498 499 tests += Iter(n, """, 500 const T%s& v%s""") 501 502 tests += """) const { 503 return PredFormatFunction%(n)s(%(es)s, %(vs)s); 504 } 505}; 506""" % DEFS 507 508 tests += """ 509// Tests for {EXPECT|ASSERT}_PRED_FORMAT%(n)s. 510 511class Predicate%(n)sTest : public testing::Test { 512 protected: 513 virtual void SetUp() { 514 expected_to_finish_ = true; 515 finished_ = false;""" % DEFS 516 517 tests += """ 518 """ + Iter(n, 'n%s_ = ') + """0; 519 } 520""" 521 522 tests += """ 523 virtual void TearDown() { 524 // Verifies that each of the predicate's arguments was evaluated 525 // exactly once.""" 526 527 tests += ''.join([""" 528 EXPECT_EQ(1, n%s_) << 529 "The predicate assertion didn't evaluate argument %s " 530 "exactly once.";""" % (i, i + 1) for i in OneTo(n)]) 531 532 tests += """ 533 534 // Verifies that the control flow in the test function is expected. 535 if (expected_to_finish_ && !finished_) { 536 FAIL() << "The predicate assertion unexpactedly aborted the test."; 537 } else if (!expected_to_finish_ && finished_) { 538 FAIL() << "The failed predicate assertion didn't abort the test " 539 "as expected."; 540 } 541 } 542 543 // true iff the test function is expected to run to finish. 544 static bool expected_to_finish_; 545 546 // true iff the test function did run to finish. 547 static bool finished_; 548""" % DEFS 549 550 tests += Iter(n, """ 551 static int n%s_;""") 552 553 tests += """ 554}; 555 556bool Predicate%(n)sTest::expected_to_finish_; 557bool Predicate%(n)sTest::finished_; 558""" % DEFS 559 560 tests += Iter(n, """int Predicate%%(n)sTest::n%s_; 561""") % DEFS 562 563 tests += """ 564typedef Predicate%(n)sTest EXPECT_PRED_FORMAT%(n)sTest; 565typedef Predicate%(n)sTest ASSERT_PRED_FORMAT%(n)sTest; 566typedef Predicate%(n)sTest EXPECT_PRED%(n)sTest; 567typedef Predicate%(n)sTest ASSERT_PRED%(n)sTest; 568""" % DEFS 569 570 def GenTest(use_format, use_assert, expect_failure, 571 use_functor, use_user_type): 572 """Returns the test for a predicate assertion macro. 573 574 Args: 575 use_format: true iff the assertion is a *_PRED_FORMAT*. 576 use_assert: true iff the assertion is a ASSERT_*. 577 expect_failure: true iff the assertion is expected to fail. 578 use_functor: true iff the first argument of the assertion is 579 a functor (as opposed to a function) 580 use_user_type: true iff the predicate functor/function takes 581 argument(s) of a user-defined type. 582 583 Example: 584 585 GenTest(1, 0, 0, 1, 0) returns a test that tests the behavior 586 of a successful EXPECT_PRED_FORMATn() that takes a functor 587 whose arguments have built-in types.""" 588 589 if use_assert: 590 assrt = 'ASSERT' # 'assert' is reserved, so we cannot use 591 # that identifier here. 592 else: 593 assrt = 'EXPECT' 594 595 assertion = assrt + '_PRED' 596 597 if use_format: 598 pred_format = 'PredFormat' 599 assertion += '_FORMAT' 600 else: 601 pred_format = 'Pred' 602 603 assertion += '%(n)s' % DEFS 604 605 if use_functor: 606 pred_format_type = 'functor' 607 pred_format += 'Functor%(n)s()' 608 else: 609 pred_format_type = 'function' 610 pred_format += 'Function%(n)s' 611 if not use_format: 612 if use_user_type: 613 pred_format += 'Bool' 614 else: 615 pred_format += 'Int' 616 617 test_name = pred_format_type.title() 618 619 if use_user_type: 620 arg_type = 'user-defined type (Bool)' 621 test_name += 'OnUserType' 622 if expect_failure: 623 arg = 'Bool(n%s_++)' 624 else: 625 arg = 'Bool(++n%s_)' 626 else: 627 arg_type = 'built-in type (int)' 628 test_name += 'OnBuiltInType' 629 if expect_failure: 630 arg = 'n%s_++' 631 else: 632 arg = '++n%s_' 633 634 if expect_failure: 635 successful_or_failed = 'failed' 636 expected_or_not = 'expected.' 637 test_name += 'Failure' 638 else: 639 successful_or_failed = 'successful' 640 expected_or_not = 'UNEXPECTED!' 641 test_name += 'Success' 642 643 # A map that defines the values used in the test template. 644 defs = DEFS.copy() 645 defs.update({ 646 'assert' : assrt, 647 'assertion' : assertion, 648 'test_name' : test_name, 649 'pf_type' : pred_format_type, 650 'pf' : pred_format, 651 'arg_type' : arg_type, 652 'arg' : arg, 653 'successful' : successful_or_failed, 654 'expected' : expected_or_not, 655 }) 656 657 test = """ 658// Tests a %(successful)s %(assertion)s where the 659// predicate-formatter is a %(pf_type)s on a %(arg_type)s. 660TEST_F(%(assertion)sTest, %(test_name)s) {""" % defs 661 662 indent = (len(assertion) + 3)*' ' 663 extra_indent = '' 664 665 if expect_failure: 666 extra_indent = ' ' 667 if use_assert: 668 test += """ 669 expected_to_finish_ = false; 670 EXPECT_FATAL_FAILURE({ // NOLINT""" 671 else: 672 test += """ 673 EXPECT_NONFATAL_FAILURE({ // NOLINT""" 674 675 test += '\n' + extra_indent + """ %(assertion)s(%(pf)s""" % defs 676 677 test = test % defs 678 test += Iter(n, ',\n' + indent + extra_indent + '%(arg)s' % defs) 679 test += ');\n' + extra_indent + ' finished_ = true;\n' 680 681 if expect_failure: 682 test += ' }, "");\n' 683 684 test += '}\n' 685 return test 686 687 # Generates tests for all 2**6 = 64 combinations. 688 tests += ''.join([GenTest(use_format, use_assert, expect_failure, 689 use_functor, use_user_type) 690 for use_format in [0, 1] 691 for use_assert in [0, 1] 692 for expect_failure in [0, 1] 693 for use_functor in [0, 1] 694 for use_user_type in [0, 1] 695 ]) 696 697 return tests 698 699 700def UnitTestPostamble(): 701 """Returns the postamble for the tests.""" 702 703 return '' 704 705 706def GenerateUnitTest(n): 707 """Returns the tests for up-to n-ary predicate assertions.""" 708 709 GenerateFile(UNIT_TEST, 710 UnitTestPreamble() 711 + ''.join([TestsForArity(i) for i in OneTo(n)]) 712 + UnitTestPostamble()) 713 714 715def _Main(): 716 """The entry point of the script. Generates the header file and its 717 unit test.""" 718 719 if len(sys.argv) != 2: 720 print __doc__ 721 print 'Author: ' + __author__ 722 sys.exit(1) 723 724 n = int(sys.argv[1]) 725 GenerateHeader(n) 726 GenerateUnitTest(n) 727 728 729if __name__ == '__main__': 730 _Main() 731