1
1
openmpi/test/class/opal_bitmap.c
2016-11-22 15:03:20 -08:00

335 строки
8.2 KiB
C

/*
* Testcase for bitmap
*/
#include "opal_config.h"
#include <stdio.h>
#include "support.h"
#include "opal/class/opal_bitmap.h"
#include "opal/constants.h"
#define BSIZE 26
#define OPAL_INVALID_BIT -1
#define ERR_CODE -2
#define PRINT_VALID_ERR \
fprintf(error_out, "================================ \n"); \
fprintf(error_out, "This is suppossed to throw error \n"); \
fprintf(error_out, "================================ \n")
static void test_bitmap_set(opal_bitmap_t *bm);
static void test_bitmap_clear(opal_bitmap_t *bm);
static void test_bitmap_is_set(opal_bitmap_t *bm);
static void test_bitmap_clear_all(opal_bitmap_t *bm);
static void test_bitmap_set_all(opal_bitmap_t *bm);
static void test_bitmap_find_and_set(opal_bitmap_t *bm);
static int set_bit(opal_bitmap_t *bm, int bit);
static int clear_bit(opal_bitmap_t *bm, int bit);
static int is_set_bit(opal_bitmap_t *bm, int bit);
static int clear_all(opal_bitmap_t *bm);
static int set_all(opal_bitmap_t *bm);
static int find_and_set(opal_bitmap_t *bm, int bit);
#define WANT_PRINT_BITMAP 0
#if WANT_PRINT_BITMAP
static void print_bitmap(opal_bitmap_t *bm);
#endif
static FILE *error_out=NULL;
int main(int argc, char *argv[])
{
/* Local variables */
opal_bitmap_t bm;
int err;
/* Perform overall test initialization */
test_init("opal_bitmap_t");
#ifdef STANDALONE
error_out = stderr;
#else
error_out = fopen( "./opal_bitmap_test_out.txt", "w" );
if( error_out == NULL ) error_out = stderr;
#endif
/* Initialize bitmap */
OBJ_CONSTRUCT(&bm, opal_bitmap_t);
PRINT_VALID_ERR;
err = opal_bitmap_init(NULL, 2);
if (err == OPAL_ERR_BAD_PARAM)
fprintf(error_out, "ERROR: Initialization of bitmap failed\n\n");
PRINT_VALID_ERR;
err = opal_bitmap_init(&bm, -1);
if (err == OPAL_ERR_BAD_PARAM)
fprintf(error_out, "ERROR: Initialization of bitmap failed \n\n");
err = opal_bitmap_init(&bm, BSIZE);
if (0 > err) {
fprintf(error_out, "Error in bitmap create -- aborting \n");
exit(-1);
}
fprintf(error_out, "\nTesting bitmap set... \n");
test_bitmap_set(&bm);
fprintf(error_out, "\nTesting bitmap clear ... \n");
test_bitmap_clear(&bm);
fprintf(error_out, "\nTesting bitmap is_set ... \n");
test_bitmap_is_set(&bm);
fprintf(error_out, "\nTesting bitmap clear_all... \n");
test_bitmap_clear_all(&bm);
fprintf(error_out, "\nTesting bitmap set_all... \n");
test_bitmap_set_all(&bm);
fprintf(error_out, "\nTesting bitmap find_and_set... \n");
test_bitmap_find_and_set(&bm);
fprintf(error_out, "\n~~~~~~ Testing complete ~~~~~~ \n\n");
test_finalize();
#ifndef STANDALONE
fclose(error_out);
#endif
return 0;
}
void test_bitmap_set(opal_bitmap_t *bm) {
int result=0;
/* start of bitmap and boundaries */
set_bit(bm, 0);
set_bit(bm, 1);
set_bit(bm, 7);
set_bit(bm, 8);
/* middle of bitmap */
set_bit(bm, 24);
/* end of bitmap initial size */
set_bit(bm, 31);
set_bit(bm, 32);
/* beyond bitmap -- this is valid */
set_bit(bm, 44);
set_bit(bm, 82);
/* invalid bit */
PRINT_VALID_ERR;
result = set_bit(bm, -1);
TEST_AND_REPORT(result, ERR_CODE,"opal_bitmap_set_bit");
}
void test_bitmap_clear(opal_bitmap_t *bm) {
int result=0;
/* Valid set bits */
clear_bit(bm, 29);
clear_bit(bm, 31);
clear_bit(bm, 33);
clear_bit(bm, 32);
clear_bit(bm, 0);
/* invalid bit */
PRINT_VALID_ERR;
result = clear_bit(bm, -1);
TEST_AND_REPORT(result, ERR_CODE,"opal_bitmap_clear_bit");
PRINT_VALID_ERR;
result = clear_bit(bm, 142);
TEST_AND_REPORT(result, ERR_CODE,"opal_bitmap_clear_bit");
}
void test_bitmap_is_set(opal_bitmap_t *bm)
{
int result=0;
/* First set some bits */
test_bitmap_set(bm);
is_set_bit(bm, 0);
is_set_bit(bm, 1);
is_set_bit(bm, 31);
is_set_bit(bm, 32);
result = is_set_bit(bm, 1122);
TEST_AND_REPORT(result,0,"opal_bitmap_is_set_bit");
is_set_bit(bm, -33);
TEST_AND_REPORT(result,0,"opal_bitmap_is_set_bit");
is_set_bit(bm, -1);
TEST_AND_REPORT(result,0,"opal_bitmap_is_set_bit");
}
void test_bitmap_find_and_set(opal_bitmap_t *bm)
{
int bsize;
int result=0;
opal_bitmap_clear_all_bits(bm);
result = find_and_set(bm, 0);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
result = find_and_set(bm, 1);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
result = find_and_set(bm, 2);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
result = find_and_set(bm, 3);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
result = opal_bitmap_set_bit(bm, 5);
result = find_and_set(bm, 4);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
result = opal_bitmap_set_bit(bm, 6);
result = opal_bitmap_set_bit(bm, 7);
/* Setting beyond a char boundary */
result = find_and_set(bm, 8);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
opal_bitmap_set_bit(bm, 9);
result = find_and_set(bm, 10);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
/* Setting beyond the current size of bitmap */
opal_bitmap_set_all_bits(bm);
bsize = opal_bitmap_size(bm);
result = find_and_set(bm, bsize);
TEST_AND_REPORT(result, 0, "opal_bitmap_find_and_set_first_unset_bit");
}
void test_bitmap_clear_all(opal_bitmap_t *bm)
{
int result = clear_all(bm);
TEST_AND_REPORT(result, 0, " error in opal_bitmap_clear_all_bits");
}
void test_bitmap_set_all(opal_bitmap_t *bm)
{
int result = set_all(bm);
TEST_AND_REPORT(result, 0, " error in opal_bitmap_set_ala_bitsl");
}
int set_bit(opal_bitmap_t *bm, int bit)
{
int err = opal_bitmap_set_bit(bm, bit);
if (err != 0 || !opal_bitmap_is_set_bit(bm, bit)) {
fprintf(error_out, "ERROR: set_bit for bit = %d\n\n", bit);
return ERR_CODE;
}
return 0;
}
int clear_bit(opal_bitmap_t *bm, int bit)
{
int err = opal_bitmap_clear_bit(bm, bit);
if ((err != 0) || opal_bitmap_is_set_bit(bm, bit)) {
fprintf(error_out, "ERROR: clear_bit for bit = %d \n\n", bit);
return ERR_CODE;
}
return 0;
}
int is_set_bit(opal_bitmap_t *bm, int bit)
{
bool result = opal_bitmap_is_set_bit(bm, bit);
if (result) {
if (bit < 0) {
fprintf(error_out, "ERROR: is_set_bit for bit = %d \n\n",bit);
return ERR_CODE;
}
return 0;
}
if (!result) {
if (0 <= bit && bit <= bm->array_size && !opal_bitmap_is_set_bit(bm, bit)) {
fprintf(error_out, "ERROR: is_set_bit for bit = %d \n\n",bit);
return ERR_CODE;
}
return 0;
}
return 0;
}
int find_and_set(opal_bitmap_t *bm, int bit)
{
int ret, pos;
/* bit here is the bit that should be found and set, in the top
level stub, this function will be called in sequence to test */
ret = opal_bitmap_find_and_set_first_unset_bit(bm, &pos);
if (ret != OPAL_SUCCESS) return ret;
if (pos != bit) {
fprintf(error_out, "ERROR: find_and_set: expected to find_and_set %d\n\n",
bit);
return ERR_CODE;
}
return 0;
}
int clear_all(opal_bitmap_t *bm)
{
int i;
if (OPAL_SUCCESS != opal_bitmap_clear_all_bits(bm)) {
return ERR_CODE;
}
for (i = 0; i < bm->array_size; ++i)
if (bm->bitmap[i] != 0) {
fprintf(error_out, "ERROR: clear_all for bitmap array entry %d\n\n",
i);
return ERR_CODE;
}
return 0;
}
int set_all(opal_bitmap_t *bm)
{
int i;
if (OPAL_SUCCESS != opal_bitmap_set_all_bits(bm)) {
return ERR_CODE;
}
for (i = 0; i < bm->array_size; ++i)
if (bm->bitmap[i] != 0xffffffffffffffffUL) {
fprintf(error_out, "ERROR: set_all for bitmap arry entry %d\n\n", i);
return ERR_CODE;
}
return 0;
}
#if WANT_PRINT_BITMAP
void print_bitmap(opal_bitmap_t *bm)
{
/* Accessing the fields within the structure, since its not an
opaque structure */
int i;
for (i = 0; i < bm->array_size; ++i) {
fprintf(error_out, "---\n bitmap[%d] = %x \n---\n\n", i,
(bm->bitmap[i] & 0xff));
}
fprintf(error_out, "========================= \n");
return;
}
#endif