Blame view

3rdparty/boost_1_81_0/libs/python/test/slice.cpp 2.88 KB
73ef4ff3   Hu Chunming   提交三方库
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
  #include <boost/python.hpp>
  #include <boost/python/slice.hpp>
  #include <boost/python/str.hpp>
  #include <vector>
  
  // Copyright (c) 2004 Jonathan Brandmeyer
  //  Use, modification and distribution are subject to the
  //  Boost Software License, Version 1.0. (See accompanying file
  //  LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  
  using namespace boost::python;
  
  #if BOOST_WORKAROUND(__SUNPRO_CC, BOOST_TESTED_AT(0x580)) || BOOST_WORKAROUND(BOOST_MSVC, BOOST_TESTED_AT(1500))
  # define make_tuple boost::python::make_tuple
  #endif 
  
  // These checks are only valid under Python 2.3
  // (rich slicing wasn't supported for builtins under Python 2.2)
  bool check_string_rich_slice()
  {
      object s("hello, world");
  
      // default slice
      if (s[slice()] != "hello, world")
          return false;
  
      // simple reverse
      if (s[slice(_,_,-1)] != "dlrow ,olleh")
          return false;
  
      // reverse with mixed-sign offsets
      if (s[slice(-6,1,-1)] != " ,oll")
          return false;
  
      // all of the object.cpp check_string_slice() checks should work
      // with the form that omits the step argument.
      if (s[slice(_,-3)] != "hello, wo")
          return false;
      if (s[slice(-3,_)] != "rld")
          return false;
      if (", " != s[slice(5,7)])
          return false;
  
      return s[slice(2,-1)][slice(1,-1)]  == "lo, wor";
  }
  
  // Tried to get more info into the error message (actual array
  // contents) but Numeric complains that treating an array as a boolean
  // value doesn't make any sense.
  #define ASSERT_EQUAL( e1, e2 ) \
      if (!all((e1) == (e2)))                                                             \
          return "assertion failed: " #e1 " == " #e2 "\nLHS:\n%s\nRHS:\n%s" % make_tuple(e1,e2);    \
  else
  
  // Verify functions accepting a slice argument can be called
  bool accept_slice( slice) { return true; }
  
  #if BOOST_WORKAROUND( BOOST_MSVC, BOOST_TESTED_AT(1400)) \
      || BOOST_WORKAROUND( BOOST_INTEL_WIN, == 710)
  int check_slice_get_indices(slice index);
  #endif
  int check_slice_get_indices(
  #if !BOOST_WORKAROUND(__SUNPRO_CC, BOOST_TESTED_AT(0x590))
      const
  #endif 
      slice index)
  {
      // A vector of integers from [-5, 5].
      std::vector<int> coll(11);
      typedef std::vector<int>::iterator coll_iterator;
      
      for (coll_iterator i = coll.begin(); i != coll.end(); ++i) {
          *i = i - coll.begin() - 5;
      }
      
      slice::range<std::vector<int>::iterator> bounds;
      try {
          bounds = index.get_indices(coll.begin(), coll.end());
      }
      catch (std::invalid_argument) {
          return 0;
      }
      int sum = 0;
      while (bounds.start != bounds.stop) {
          sum += *bounds.start;
          std::advance( bounds.start, bounds.step);
      }
      sum += *bounds.start;
      return sum;
  }
  
  
  BOOST_PYTHON_MODULE(slice_ext)
  {
      def( "accept_slice", accept_slice);
      def( "check_string_rich_slice", check_string_rich_slice);
      def( "check_slice_get_indices", check_slice_get_indices);
  }