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
|
/* Boost.MultiIndex test for special set operations.
*
* Copyright 2003-2008 Joaquin M Lopez Munoz.
* Distributed under 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)
*
* See http://www.boost.org/libs/multi_index for library home page.
*/
#include "test_special_set_ops.hpp"
#include <boost/config.hpp> /* keep it first to prevent nasty warns in MSVC */
#include <algorithm>
#include <sstream>
#include "pre_multi_index.hpp"
#include "employee.hpp"
#include <boost/test/test_tools.hpp>
using namespace boost::multi_index;
static int string_to_int(const std::string& str)
{
std::istringstream iss(str);
int res;
iss>>res;
return res;
}
struct comp_int_string
{
bool operator()(int x,const std::string& y)const
{
return x<string_to_int(y);
}
bool operator()(const std::string& x,int y)const
{
return string_to_int(x)<y;
}
};
struct hash_string_as_int
{
int operator()(const std::string& x)const
{
return boost::hash<int>()(string_to_int(x));
}
};
struct eq_string_int
{
bool operator()(int x,const std::string& y)const
{
return x==string_to_int(y);
}
bool operator()(const std::string& x,int y)const
{
return operator()(y,x);
}
};
void test_special_set_ops()
{
employee_set es;
es.insert(employee(0,"Joe",31,1123));
es.insert(employee(1,"Robert",27,5601));
es.insert(employee(2,"John",40,7889));
es.insert(employee(3,"Albert",20,9012));
es.insert(employee(4,"John",57,1002));
std::pair<employee_set_by_ssn::iterator,employee_set_by_ssn::iterator> p=
get<ssn>(es).equal_range(
"7889",hash_string_as_int(),eq_string_int());
BOOST_CHECK(std::distance(p.first,p.second)==1&&(p.first)->id==2);
BOOST_CHECK(
get<ssn>(es).count(
"5601",hash_string_as_int(),eq_string_int())==1);
BOOST_CHECK(
get<ssn>(es).find(
"1123",hash_string_as_int(),eq_string_int())->name=="Joe");
BOOST_CHECK(
std::distance(
get<age>(es).lower_bound("27",comp_int_string()),
get<age>(es).upper_bound("40",comp_int_string()))==3);
BOOST_CHECK(es.count(2,employee::comp_id())==1);
BOOST_CHECK(es.count(5,employee::comp_id())==0);
}
|
