IntervalIndex#
-
class numpy::IntervalIndex#
numpy C++ class.
Example#
#include <numpy/np_ndarray.h>
using namespace numpy;
// Use IntervalIndex
IntervalIndex obj;
// ... operations ...
Constructors#
Signature |
Location |
Example |
|---|---|---|
|
df_interval_index.h:124 |
|
|
df_interval_index.h:133 |
|
|
df_interval_index.h:142 |
|
|
df_interval_index.h:150 |
Construction#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
static IntervalIndex |
df_interval_index.h:175 |
|
|
static IntervalIndex |
df_interval_index.h:188 |
|
|
static IntervalIndex |
df_interval_index.h:201 |
|
|
static IntervalIndex |
df_interval_index.h:213 |
|
|
static IntervalIndex |
df_interval_index.h:225 |
Indexing / Selection#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
df_interval_index.h:1006 |
||
|
IntervalIndex |
df_interval_index.h:1279 |
|
|
size_t |
df_interval_index.h:932 |
|
|
std::string |
df_interval_index.h:475 |
|
|
std::pair<T, T> |
df_interval_index.h:652 |
|
|
IntervalIndex |
df_interval_index.h:1206 |
|
|
size_t |
df_interval_index.h:1041 |
|
|
IntervalIndex |
df_interval_index.h:1179 |
Shape Manipulation#
Statistics#
Sorting#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
IntervalIndex |
df_interval_index.h:1070 |
|
|
std::pair<IntervalIndex, numpy::NDArray<numpy::int64>> |
df_interval_index.h:1081 |
Math Operations#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
IntervalIndex |
df_interval_index.h:1249 |
Logical#
I/O#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
IntervalIndex |
df_interval_index.h:826 |
|
|
FrameData |
df_interval_index.h:1468 |
|
|
std::vector<std::pair<T, T>> |
df_interval_index.h:838 |
|
|
SeriesData |
df_interval_index.h:1535 |
|
|
std::string |
df_interval_index.h:441 |
|
|
std::vector<std::optional<std::pair<T, T>>> |
df_interval_index.h:416 |
|
|
std::vector<std::optional<std::pair<T, T>>> |
df_interval_index.h:858 |
Joining / Splitting#
Type Handling#
Type Checking#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
bool |
df_interval_index.h:1300 |
|
|
bool |
df_interval_index.h:595 |
|
|
bool |
df_interval_index.h:602 |
|
|
BooleanArray |
df_interval_index.h:309 |
|
|
bool |
df_interval_index.h:609 |
|
|
bool |
df_interval_index.h:616 |
|
|
bool |
df_interval_index.h:588 |
|
|
bool |
df_interval_index.h:288 |
|
|
bool |
df_interval_index.h:336 |
|
|
bool |
df_interval_index.h:623 |
|
|
bool |
df_interval_index.h:630 |
|
|
bool |
df_interval_index.h:321 |
|
|
bool |
df_interval_index.h:295 |
Other Methods#
Signature |
Return Type |
Location |
Example |
|---|---|---|---|
|
numpy::int64 |
df_interval_index.h:570 |
|
|
numpy::int64 |
df_interval_index.h:563 |
|
|
std::optional<std::pair<T, T>> |
df_interval_index.h:887 |
|
|
numpy::NDArray<numpy::int64> |
df_interval_index.h:906 |
|
|
std::unique_ptr<IndexBase> |
df_interval_index.h:434 |
|
|
IntervalClosed |
df_interval_index.h:274 |
|
|
std::string |
df_interval_index.h:281 |
|
|
void |
df_interval_index.h:54 |
|
|
BooleanArray |
df_interval_index.h:353 |
|
|
BooleanArray |
df_interval_index.h:363 |
|
|
numpy::NDArray<T> |
df_interval_index.h:698 |
|
|
IntervalIndex |
df_interval_index.h:1291 |
|
|
std::vector<std::string> |
df_interval_index.h:867 |
|
|
std::unordered_map<GroupKey, std::vector<size_t>> |
df_interval_index.h:1340 |
|
|
std::unordered_map<GroupT, std::vector<size_t>> |
df_interval_index.h:1370 |
|
|
bool |
df_interval_index.h:581 |
|
|
numpy::NDArray<numpy::int64> |
df_interval_index.h:1007 |
|
|
numpy::NDArray<numpy::int64> |
df_interval_index.h:1115 |
|
|
IntervalIndex |
df_interval_index.h:1240 |
|
|
std::string |
df_interval_index.h:427 |
|
|
IntervalIndex |
df_interval_index.h:1409 |
|
|
numpy::NDArray<T> |
df_interval_index.h:241 |
|
|
numpy::NDArray<T> |
df_interval_index.h:263 |
|
|
IntervalIndex |
df_interval_index.h:1156 |
|
|
size_t |
df_interval_index.h:643 |
|
|
FloatingArray<numpy::float64> |
df_interval_index.h:256 |
|
|
numpy::NDArray<numpy::bool_> |
df_interval_index.h:1008 |
|
|
IntervalArray<T> |
df_interval_index.h:689 |
|
|
IntervalArray<T> |
df_interval_index.h:775 |
|
|
IntervalArray<T> |
df_interval_index.h:795 |
|
|
IntervalArray<T> |
df_interval_index.h:1099 |
|
|
IntervalArray<T> |
df_interval_index.h:1142 |
|
|
IntervalArray<T> |
df_interval_index.h:1169 |
|
|
IntervalArray<T> |
df_interval_index.h:1196 |
|
|
IntervalArray<T> |
df_interval_index.h:1223 |
|
|
IntervalArray<T> |
df_interval_index.h:1269 |
|
|
BooleanArray |
df_interval_index.h:374 |
|
|
BooleanArray |
df_interval_index.h:386 |
|
|
std::pair<IntervalIndex, numpy::NDArray<numpy::int64>> |
df_interval_index.h:1112 |
|
|
IntervalIndex |
df_interval_index.h:489 |
|
|
std::string |
df_interval_index.h:468 |
|
|
IntervalIndex |
df_interval_index.h:490 |
|
|
numpy::NDArray<T> |
df_interval_index.h:248 |
|
|
IntervalIndex |
df_interval_index.h:400 |
|
|
IntervalIndex |
df_interval_index.h:408 |
|
|
IntervalIndex |
df_interval_index.h:1232 |
|
|
IntervalIndex |
df_interval_index.h:747 |
|
|
std::vector<size_t> |
df_interval_index.h:985 |
|
|
std::pair<size_t, size_t> |
df_interval_index.h:952 |
Code Examples#
The following examples are extracted from the test suite.
item (np_test_4_all.cpp:20048)
20038/**
20039 * @file np_test_phase6.cpp
20040 * @brief Test suite for Phase 6A - HIGH PRIORITY NDArray methods and masked array utilities
20041 *
20042 * Tests:
20043 * - ndarray.nonzero() - Return indices of non-zero elements
20044 * - ndarray.tobytes() - Convert to bytes representation
20045 * - ndarray.tolist() - Convert to nested list string
20046 * - ndarray.fill() - Fill with scalar value (in-place)
20047 * - ndarray.item() - Get single element as scalar
20048 * - ndarray.itemset() - Set single element value
20049 * - ma.masked_all_like() - Create masked array with all elements masked
20050 *
20051 * Created: 2025-10-28
20052 * Status: Phase 6A Implementation - HIGH PRIORITY functions
20053 */
20054
20055#include "../numpy/np_ndarray.h"
20056#include "../numpy/np_masked_array.h"
20057#include <iostream>
putmask (np_test_5_all.cpp:4659)
4649 mask.setElementAt({ 1 }, false);
4650 mask.setElementAt({ 2 }, true);
4651 mask.setElementAt({ 3 }, false);
4652 mask.setElementAt({ 4 }, true);
4653
4654 // Create value array
4655 numpy::NDArray<int32_t> values({ 1 });
4656 values.setElementAt({ 0 }, 99);
4657
4658 // Apply putmask
4659 numpy::putmask(arr, mask, values);
4660
4661 // Expected: [99, 2, 99, 4, 99]
4662 if (arr.getElementAt({ 0 }) != 99 || arr.getElementAt({ 1 }) != 2 ||
4663 arr.getElementAt({ 2 }) != 99 || arr.getElementAt({ 3 }) != 4 ||
4664 arr.getElementAt({ 4 }) != 99) {
4665 std::cout << " [FAIL] : in np_test_putmask() : Incorrect result";
4666 throw std::runtime_error("Test failed");
4667 }
4668
4669 // Test with multiple values
searchsorted (np_test_2_all.cpp:8204)
8194 // Test missing NumPy functions
8195 void test_numpy_functions() {
8196 std::cout << "========= NumPy Functions Test ==========================";
8197
8198 // Test searchsorted
8199 {
8200 std::vector<int> sorted_arr = { 1, 3, 5, 7, 9 };
8201 std::vector<int> values = { 2, 6, 8 };
8202
8203 auto result = numpy::searchsorted(sorted_arr, values, "left");
8204 std::vector<size_t> expected = { 1, 3, 4 };
8205
8206 if (result != expected) {
8207 std::cout << " searchsorted (left): -> [FAIL]";
8208 throw std::runtime_error("searchsorted (left): FAILED - incorrect result indices");
8209 }
8210 // std::cout << "[OK] searchsorted (left): PASSED" << std::endl;
8211
8212 auto result_right = numpy::searchsorted(sorted_arr, values, "right");
8213 std::vector<size_t> expected_right = { 1, 3, 4 };
where (np_test_3_all.cpp:15253)
15243 auto y_data = createInt32Array({ 4 }, 0);
15244 y_data.setElementAt({ 0 }, 100);
15245 y_data.setElementAt({ 1 }, 200);
15246 y_data.setElementAt({ 2 }, 300);
15247 y_data.setElementAt({ 3 }, 400);
15248
15249 auto y_mask = createBoolArray({ 4 }, false);
15250 MaskedArray<int32> y(y_data, y_mask, -999);
15251
15252 // Test where() selection
15253 auto result = where(condition, x, y);
15254
15255 if (!(result.getElementAt({ 0 }) == 10)) {
15256 std::string description = std::string("test_where_selection():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(result.getElementAt({ 0 }) == 10)";
15257 std::cout << std::string("[FAIL] ") + description << std::endl;
15258 throw std::runtime_error(description);
15259 }
15260 if (!(result.getElementAt({ 1 }) == 200)) {
15261 std::string description = std::string("test_where_selection():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(result.getElementAt({ 1 }) == 200)";
15262 std::cout << std::string("[FAIL] ") + description << std::endl;
ravel (np_test_1_all.cpp:4082)
4072 std::cout << "Testing ravel function...\n";
4073
4074 // Test with 2-D array
4075 auto arr_2d = createFloat64Array({ 2, 3 });
4076 for (size_t i = 0; i < 2; ++i) {
4077 for (size_t j = 0; j < 3; ++j) {
4078 arr_2d.setElementAt({ i, j }, static_cast<double>(i * 3 + j));
4079 }
4080 }
4081
4082 auto raveled = ravel(arr_2d);
4083 if (!(raveled.getShape() == std::vector<size_t>({ 6 }))) {
4084 std::string description = std::string("testRavelFunction():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(raveled.getShape() == std::vector<size_t>({ 6 }))";
4085 std::cout << std::string("[FAIL] ") + description << std::endl;
4086 throw std::runtime_error(description);
4087 }
4088
4089 for (size_t i = 0; i < 6; ++i) {
4090 if (!(isApproxEqualExt(raveled.getElementAt({ i }), static_cast<double>(i)))) {
4091 std::string description = std::string("testRavelFunction():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(isApproxEqualExt(raveled.getElementAt({ i }), static_cast<double>(i)))";
4092 std::cout << std::string("[FAIL] ") + description << std::endl;
transpose (np_test_2_all.cpp:4973)
4963 }
4964
4965 /**
4966 * Test matrix properties and methods
4967 */
4968 void testMatrixProperties() {
4969 std::cout << "========= testMatrixProperties =======================";
4970
4971 // Test transpose
4972 numpy::Matrix<double> m("1 2 3; 4 5 6");
4973 auto mt = m.transpose();
4974 assert_test(mt.rows() == 3, "Transpose rows");
4975 assert_test(mt.cols() == 2, "Transpose cols");
4976 assert_test(std::abs(mt(0, 1) - 4.0) < 1e-10, "Transpose element");
4977 assert_test(std::abs(mt(2, 0) - 3.0) < 1e-10, "Transpose element");
4978
4979 // Test trace for square matrix
4980 numpy::Matrix<double> square("1 2; 3 4");
4981 double tr = square.trace();
4982 assert_test(std::abs(tr - 5.0) < 1e-10, "Matrix trace");
max (np_test_1_all.cpp:7274)
7264 if (sizeof(uintp) == sizeof(void*)) {
7265 // std::cout << " -> uintp size matches pointer size";
7266 } else {
7267 // std::cout << " ✗ uintp size doesn't match pointer size" << std::endl;
7268 }
7269
7270 // Test range limits
7271 // std::cout << "Range Information:" << std::endl;
7272 // std::cout << " intp min: " << std::numeric_limits<intp>::min() << std::endl;
7273 // std::cout << " intp max: " << std::numeric_limits<intp>::max() << std::endl;
7274 // std::cout << " uintp max: " << std::numeric_limits<uintp>::max() << std::endl;
7275 // std::cout << " longdouble digits: " << std::numeric_limits<longdouble>::digits << std::endl;
7276
7277 std::cout << " -> tests passed" << std::endl;
7278}
7279
7280void testComplexArithmeticExtendedTypes() {
7281 std::cout << "========= testComplexArithmeticExtendedTypes =======================";
7282
7283 clongdouble c1(3.0L, 4.0L); // 3 + 4i
min (np_test_1_all.cpp:2350)
2340 if (i % 3 == 0) {
2341 large_array.setElementAt({i}, object_(static_cast<int>(i)));
2342 } else if (i % 3 == 1) {
2343 large_array.setElementAt({i}, object_(static_cast<double>(i) * 0.5));
2344 } else {
2345 large_array.setElementAt({i}, object_(std::string("str") + std::to_string(i)));
2346 }
2347 }
2348
2349 // Verify pattern
2350 for (size_t i = 0; i < std::min(large_size, size_t(100)); ++i) { // Check first 100
2351 object_ obj = large_array.getElementAt({i});
2352 if (i % 3 == 0) {
2353 if (!(obj.is_type<int>())) {
2354 std::string description = std::string("testArrayEdgeCases():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(obj.is_type<int>())";
2355 std::cout << std::string("[FAIL] ") + description << std::endl;
2356 throw std::runtime_error(description);
2357 }
2358 } else if (i % 3 == 1) {
2359 if (!(obj.is_type<double>())) {
2360 std::string description = std::string("unknown_function():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(obj.is_type<double>())";
sort (np_test_1_all.cpp:6258)
6248 std::vector<int> data = gen.generate(100, DataPattern::FEW_UNIQUE);
6249
6250 if (!(data.size() == 100)) {
6251 std::string description = std::string("test_data_generator_few_uniqueBenchmarkSorting():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(data.size() == 100)";
6252 std::cout << std::string("[FAIL] ") + description << std::endl;
6253 throw std::runtime_error(description);
6254 }
6255
6256 // Count unique values
6257 std::vector<int> sorted_copy = data;
6258 std::sort(sorted_copy.begin(), sorted_copy.end());
6259 auto unique_end = std::unique(sorted_copy.begin(), sorted_copy.end());
6260 size_t unique_count = std::distance(sorted_copy.begin(), unique_end);
6261
6262 // Should have significantly fewer unique values than total elements
6263 if (!(unique_count < data.size() / 2)) {
6264 std::string description = std::string("test_data_generator_few_uniqueBenchmarkSorting():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(unique_count < data.size() / 2)";
6265 std::cout << std::string("[FAIL] ") + description << std::endl;
6266 throw std::runtime_error(description);
6267 }
round (np_test_1_all.cpp:23769)
23759 throw std::runtime_error(description);
23760 }
23761 if (!(power_scalar_result.getShape() == array.getShape())) {
23762 std::string description = std::string("testMathFunctionSignatures():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(power_scalar_result.getShape() == array.getShape())";
23763 std::cout << std::string("[FAIL] ") + description << std::endl;
23764 throw std::runtime_error(description);
23765 }
23766 // std::cout << "[OK] Power functions have correct signatures\n";
23767
23768 // Test rounding functions with new decimals parameter
23769 auto round_result = round(array); // Default decimals=0
23770 auto round_decimals_result = round(array, 2); // With decimals
23771 if (!(round_result.getShape() == array.getShape())) {
23772 std::string description = std::string("testMathFunctionSignatures():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(round_result.getShape() == array.getShape())";
23773 std::cout << std::string("[FAIL] ") + description << std::endl;
23774 throw std::runtime_error(description);
23775 }
23776 if (!(round_decimals_result.getShape() == array.getShape())) {
23777 std::string description = std::string("testMathFunctionSignatures():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(round_decimals_result.getShape() == array.getShape())";
23778 std::cout << std::string("[FAIL] ") + description << std::endl;
23779 throw std::runtime_error(description);
all (np_test_4_all.cpp:23928)
23918 }
23919 }
23920 }
23921
23922 auto result = numpy::einsum<numpy::float64>("ijk->ik", {A});
23923
23924 if (result.getShape()[0] != 2 || result.getShape()[1] != 2) {
23925 throw std::runtime_error("einsum partial sum shape wrong");
23926 }
23927
23928 // Sum over j: 1+2+3 = 6 for all (i,k) positions
23929 if (std::abs(result.getElementAt({0, 0}) - 6.0) > 1e-10 ||
23930 std::abs(result.getElementAt({1, 1}) - 6.0) > 1e-10) {
23931 throw std::runtime_error("einsum partial sum values wrong");
23932 }
23933
23934 // std::cout << " OK: Partial sum: 'ijk->ik'\n";
23935 }
23936
23937 // Test 5: Size-1 dimension handling
23938 {
any (np_test_2_all.cpp:16758)
16748 // ANY() TESTS - SCALAR RESULT
16749 // ============================================================================
16750
16751 void np_test_logic_any_scalar_all_false() {
16752 std::cout << "========= any: all false elements =======================";
16753
16754 // Create array with all false/zero elements
16755 std::vector<double> data = { 0.0, 0.0, 0.0 };
16756 numpy::NDArray<double> arr = numpy::createArrayFromVector<double>({ 3 }, data);
16757
16758 bool result = numpy::any(arr);
16759
16760 if (result != false) {
16761 std::cout << " [FAIL] : in np_test_logic_any_scalar_all_false() : expected false for all-zero array";
16762 throw std::runtime_error("np_test_logic_any_scalar_all_false failed: expected false");
16763 }
16764
16765 std::cout << " -> tests passed" << std::endl;
16766 }
16767
16768 void np_test_logic_any_scalar_all_true() {
to_numpy (np_test_5_all.cpp:21373)
21363 if (errors == 0) {
21364 std::cout << "np_test_timedelta_components -> tests passed" << std::endl;
21365 }
21366 return errors;
21367}
21368
21369// =============================================================================
21370// Test 4: Total Conversion Properties
21371// =============================================================================
21372int np_test_timedelta_total_conversions() {
21373 int errors = 0;
21374
21375 numpy::Timedelta td(1, 12, 0); // 1 day 12 hours = 1.5 days = 36 hours
21376
21377 // total_seconds
21378 {
21379 double secs = td.total_seconds();
21380 double expected = (24 + 12) * 3600.0;
21381 if (std::abs(secs - expected) > 0.0001) {
21382 std::cout << "[FAIL] np_test_timedelta_total_conversions: total_seconds expected "
21383 << expected << ", got " << secs << std::endl;
to_string (np_test_1_all.cpp:454)
444 // Modify through different views
445 view1.setElementAt({0, 0}, 100);
446 view2.setElementAt({2, 1}, 200); // This is (1, 2) in original
447 view3.setElementAt({0, 0}, 300); // This is (1, 1) in original
448
449 // std::cout << "After modifications through multiple views:" << std::endl;
450 //original.printArray();
451
452 // Verify all changes are reflected
453 if (!(original.getElementAt({0, 0}) == 100)) {
454 std::string description = std::string("testAdvancedViewLifecycle():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(original.getElementAt({0, 0}) == 100)";
455 std::cout << std::string("[FAIL] ") + description << std::endl;
456 throw std::runtime_error(description);
457 }
458 if (!(original.getElementAt({1, 2}) == 200)) {
459 std::string description = std::string("testAdvancedViewLifecycle():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(original.getElementAt({1, 2}) == 200)";
460 std::cout << std::string("[FAIL] ") + description << std::endl;
461 throw std::runtime_error(description);
462 }
463 if (!(original.getElementAt({1, 1}) == 300)) {
464 std::string description = std::string("testAdvancedViewLifecycle():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(original.getElementAt({1, 1}) == 300)";
tolist (np_test_4_all.cpp:20046)
20036} // namespace numpy_tests
20037
20038/**
20039 * @file np_test_phase6.cpp
20040 * @brief Test suite for Phase 6A - HIGH PRIORITY NDArray methods and masked array utilities
20041 *
20042 * Tests:
20043 * - ndarray.nonzero() - Return indices of non-zero elements
20044 * - ndarray.tobytes() - Convert to bytes representation
20045 * - ndarray.tolist() - Convert to nested list string
20046 * - ndarray.fill() - Fill with scalar value (in-place)
20047 * - ndarray.item() - Get single element as scalar
20048 * - ndarray.itemset() - Set single element value
20049 * - ma.masked_all_like() - Create masked array with all elements masked
20050 *
20051 * Created: 2025-10-28
20052 * Status: Phase 6A Implementation - HIGH PRIORITY functions
20053 */
20054
20055#include "../numpy/np_ndarray.h"
insert (np_test_1_all.cpp:6990)
6980}
6981
6982void testBusinessDateRangeDateTime() {
6983 std::cout << "========= testBusinessDateRangeDateTime =======================";
6984
6985 datetime64 start("2024-03-11"); // Monday
6986 datetime64 end("2024-03-22"); // Friday (2 weeks later)
6987
6988 // Create holidays set
6989 std::set<datetime64> holidays;
6990 holidays.insert(datetime64("2024-03-15")); // Friday holiday
6991
6992 auto business_dates = createBusinessDateRange(start, end, holidays);
6993 // std::cout << "Business dates from " << start.toString()
6994 // << " to " << end.toString() << " (excluding 2024-03-15):" << std::endl;
6995 //business_dates.printArray();
6996
6997 std::cout << " -> tests passed" << std::endl;
6998}
6999
7000void testMonthRangeDateTime() {
repeat (np_test_1_all.cpp:18268)
18258 // Create test array
18259 auto array = createInt32Array({ 2, 3 }, 0);
18260 for (size_t i = 0; i < 2; ++i) {
18261 for (size_t j = 0; j < 3; ++j) {
18262 array.setElementAt({ i, j }, static_cast<int32_t>(i * 3 + j + 1));
18263 }
18264 }
18265
18266 // Test repeat without axis (flattened)
18267 auto repeated_flat = repeat(array, 2);
18268 if (!(repeated_flat.getShape()[0] == 12)) {
18269 std::string description = std::string("testRepeat():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(repeated_flat.getShape()[0] == 12)";
18270 std::cout << std::string("[FAIL] ") + description << std::endl;
18271 throw std::runtime_error(description);
18272 }
18273 if (!(repeated_flat.getElementAt({ 0 }) == 1)) {
18274 std::string description = std::string("testRepeat():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(repeated_flat.getElementAt({ 0 }) == 1)";
18275 std::cout << std::string("[FAIL] ") + description << std::endl;
18276 throw std::runtime_error(description);
18277 }
astype (np_test_3_all.cpp:796)
786 auto ufunc_square = numpy::frompyfunc(square, 1, 1);
787
788 numpy::NDArray<double> arr({ 3 });
789 arr.setElementAt({ 0 }, 2.0);
790 arr.setElementAt({ 1 }, 3.0);
791 arr.setElementAt({ 2 }, 4.0);
792
793 auto result = ufunc_square(arr);
794
795 // Convert back to double to check values
796 auto typed_result = numpy::astype<double>(result);
797 if (std::abs(typed_result.getElementAt({ 0 }) - 4.0) > 1e-10 ||
798 std::abs(typed_result.getElementAt({ 1 }) - 9.0) > 1e-10 ||
799 std::abs(typed_result.getElementAt({ 2 }) - 16.0) > 1e-10) {
800 std::cout << "[FAIL] Basic frompyfunc creation failed";
801 return 1;
802 }
803
804 // std::cout << "[OK] Basic frompyfunc creation works correctly" << std::endl;
805 std::cout << " -> tests passed" << std::endl;
806 return 0;
copy (np_test_1_all.cpp:9812)
9802 //original.printArray();
9803
9804 // The modification should be at position (1,1) in original
9805 if (!(original.getElementAt({1, 1}) == 9999)) {
9806 std::string description = std::string("testSliceCopyVsViewMemoryOwnership():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(original.getElementAt({1, 1}) == 9999)";
9807 std::cout << std::string("[FAIL] ") + description << std::endl;
9808 throw std::runtime_error(description);
9809 }
9810 // std::cout << "[OK] Slice view shares memory with original";
9811
9812 // Test slice copy (should be independent)
9813 auto slice_copy = original.sliceArray({{2, 4}, {2, 4}});
9814 // std::cout << "Slice copy [2:4, 2:4]:";
9815 //slice_copy.printArray();
9816
9817 slice_copy.setElementAt({0, 0}, 7777);
9818
9819 // std::cout << "After modifying slice copy at (0,0):" << std::endl;
9820 // std::cout << "Original array:" << std::endl;
9821 //original.printArray();
9822 // std::cout << "Slice copy:" << std::endl;
view (np_test_1_all.cpp:440)
430 for (size_t i = 0; i < 3; ++i) {
431 for (size_t j = 0; j < 3; ++j) {
432 original.setElementAt({i, j}, static_cast<int32_t>(i * 3 + j));
433 }
434 }
435
436 // std::cout << "Original array:" << std::endl;
437 //original.printArray();
438
439 // Create multiple views
440 auto view1 = original.view();
441 auto view2 = original.transposeView();
442 auto view3 = original.sliceView({{1, 3}, {1, 3}});
443
444 // Modify through different views
445 view1.setElementAt({0, 0}, 100);
446 view2.setElementAt({2, 1}, 200); // This is (1, 2) in original
447 view3.setElementAt({0, 0}, 300); // This is (1, 1) in original
448
449 // std::cout << "After modifications through multiple views:" << std::endl;
450 //original.printArray();
is_numeric (np_test_1_all.cpp:7511)
7501 std::cout << " -> tests passed" << std::endl;
7502}
7503
7504void testNumericConversionStringTypes() {
7505 std::cout << "========= testNumericStringOperationsStringTypes =======================";
7506
7507 str32 num_str1("123.45");
7508 str32 num_str2("67.89");
7509 str32 not_num("hello");
7510
7511 // std::cout << "String '" << num_str1 << "' is numeric: " << is_numeric(num_str1) << std::endl;
7512 // std::cout << "String '" << not_num << "' is numeric: " << is_numeric(not_num) << std::endl;
7513
7514 if (is_numeric(num_str1)) {
7515 double val1 = to_numeric(num_str1);
7516 // std::cout << "Numeric value of '" << num_str1 << "': " << val1 << std::endl;
7517 }
7518
7519 if (is_numeric(num_str2)) {
7520 double val2 = to_numeric(num_str2);
7521 // std::cout << "Numeric value of '" << num_str2 << "': " << val2 << std::endl;
argmax (np_test_3_all.cpp:20521)
20511 auto result = numpy::argmin(arr);
20512 // Should return 1 (index of (1,5))
20513 if (result.getElementAt({0}) != 1) {
20514 throw std::runtime_error("argmin() failed: expected index 1");
20515 }
20516 std::cout << "PASSED\n";
20517}
20518
20519void test_argmax_complex128() {
20520 std::cout << " Testing argmax() with complex128... ";
20521 numpy::NDArray<complex128> arr({3});
20522 arr.setElementAt({0}, complex128(3.0, 1.0)); // index 0 - largest
20523 arr.setElementAt({1}, complex128(1.0, 5.0)); // index 1
20524 arr.setElementAt({2}, complex128(2.0, 2.0)); // index 2
20525
20526 auto result = numpy::argmax(arr);
20527 // Should return 0 (index of (3,1))
20528 if (result.getElementAt({0}) != 0) {
20529 throw std::runtime_error("argmax() failed: expected index 0");
20530 }
argmin (np_test_3_all.cpp:20506)
20496 if (min_result.getElementAt({0}).real() != 2.0 || min_result.getElementAt({0}).imag() != 1.0) {
20497 throw std::runtime_error("min() tie-breaking failed: expected (2,1)");
20498 }
20499 if (max_result.getElementAt({0}).real() != 2.0 || max_result.getElementAt({0}).imag() != 5.0) {
20500 throw std::runtime_error("max() tie-breaking failed: expected (2,5)");
20501 }
20502 std::cout << "PASSED\n";
20503}
20504
20505void test_argmin_complex128() {
20506 std::cout << " Testing argmin() with complex128... ";
20507 numpy::NDArray<complex128> arr({3});
20508 arr.setElementAt({0}, complex128(3.0, 1.0)); // index 0
20509 arr.setElementAt({1}, complex128(1.0, 5.0)); // index 1 - smallest
20510 arr.setElementAt({2}, complex128(2.0, 2.0)); // index 2
20511
20512 auto result = numpy::argmin(arr);
20513 // Should return 1 (index of (1,5))
20514 if (result.getElementAt({0}) != 1) {
20515 throw std::runtime_error("argmin() failed: expected index 1");
20516 }
clone (np_test_1_all.cpp:24942)
24932 }
24933
24934 // Test reproducibility
24935 numpy::random::PCG64DXSM rng2(11111);
24936 if (rng2.next_uint64() != val) {
24937 std::cout << " [FAIL] : in np_test_bitgen_pcg64dxsm() : not reproducible";
24938 throw std::runtime_error("np_test_bitgen_pcg64dxsm failed");
24939 }
24940
24941 // Test clone
24942 auto cloned = rng.clone();
24943 if (cloned->name() != "PCG64DXSM") {
24944 std::cout << " [FAIL] : in np_test_bitgen_pcg64dxsm() : clone name incorrect";
24945 throw std::runtime_error("np_test_bitgen_pcg64dxsm failed");
24946 }
24947
24948 std::cout << " -> tests passed" << std::endl;
24949 }
24950
24951 // ============================================================================
24952 // PHILOX TESTS
contains (np_test_2_all.cpp:10521)
10511 auto npz_file = load_npz(TEMP_DIR + "test_multiple.npz");
10512
10513 // Verify keys
10514 auto keys = npz_file.keys();
10515 if (!(keys.size() == 3)) {
10516 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(keys.size() == 3)";
10517 std::cout << std::string("[FAIL] ") + description << std::endl;
10518 throw std::runtime_error(description);
10519 }
10520 if (!(npz_file.contains("integers"))) {
10521 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(npz_file.contains(\"integers\"))";
10522 std::cout << std::string("[FAIL] ") + description << std::endl;
10523 throw std::runtime_error(description);
10524 }
10525 if (!(npz_file.contains("floats"))) {
10526 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(npz_file.contains(\"floats\"))";
10527 std::cout << std::string("[FAIL] ") + description << std::endl;
10528 throw std::runtime_error(description);
10529 }
10530 if (!(npz_file.contains("complex_nums"))) {
contains (np_test_2_all.cpp:10521)
10511 auto npz_file = load_npz(TEMP_DIR + "test_multiple.npz");
10512
10513 // Verify keys
10514 auto keys = npz_file.keys();
10515 if (!(keys.size() == 3)) {
10516 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(keys.size() == 3)";
10517 std::cout << std::string("[FAIL] ") + description << std::endl;
10518 throw std::runtime_error(description);
10519 }
10520 if (!(npz_file.contains("integers"))) {
10521 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(npz_file.contains(\"integers\"))";
10522 std::cout << std::string("[FAIL] ") + description << std::endl;
10523 throw std::runtime_error(description);
10524 }
10525 if (!(npz_file.contains("floats"))) {
10526 std::string description = std::string("testNPZOperations():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(npz_file.contains(\"floats\"))";
10527 std::cout << std::string("[FAIL] ") + description << std::endl;
10528 throw std::runtime_error(description);
10529 }
10530 if (!(npz_file.contains("complex_nums"))) {
diff (np_test_2_all.cpp:4947)
4937 // Test matrix addition and subtraction
4938 numpy::Matrix<double> a("1 2; 3 4");
4939 numpy::Matrix<double> b("5 6; 7 8");
4940
4941 auto sum = a + b;
4942 assert_test(std::abs(sum(0, 0) - 6.0) < 1e-10, "Matrix addition");
4943 assert_test(std::abs(sum(1, 1) - 12.0) < 1e-10, "Matrix addition");
4944
4945 auto diff = b - a;
4946 assert_test(std::abs(diff(0, 0) - 4.0) < 1e-10, "Matrix subtraction");
4947 assert_test(std::abs(diff(1, 1) - 4.0) < 1e-10, "Matrix subtraction");
4948
4949 // Test matrix multiplication (not element-wise!)
4950 auto product = a * b;
4951 assert_test(std::abs(product(0, 0) - 19.0) < 1e-10, "Matrix multiplication 0,0");
4952 assert_test(std::abs(product(0, 1) - 22.0) < 1e-10, "Matrix multiplication 0,1");
4953 assert_test(std::abs(product(1, 0) - 43.0) < 1e-10, "Matrix multiplication 1,0");
4954 assert_test(std::abs(product(1, 1) - 50.0) < 1e-10, "Matrix multiplication 1,1");
4955
4956 // Test scalar multiplication
format (main.cpp:22)
12int main() {
13 // Automatically log all output to temp/test_output.log
14 numpy::TestLogger logger("temp/test_output.log");
15
16 int res = 0;
17 int res1 = 0;
18 std::string resS = "";
19
20 // call all the tests
21 res1 = numpy_tests::np_test_main();
22 resS += std::format(" np_test_main: {} errors\n", res1);
23 res += res1;
24
25 std::cout << "\n------------------------- main --------------------------------------------\n";
26 std::cout << std::endl << "All tests completed. Nb errors = " << res << std::endl;
27 std::cout << "Details: \n" << resS;
28 std::cout << "\n---------------------------------------------------------------------------\n";
29 return res;
30}
join (np_test_5_all.cpp:1341)
1331 // ============================================================================
1332
1333 void np_test_string_join() {
1334 std::cout << "========= join operation =======================";
1335
1336 // Join takes a separator string and vector of CharArrays
1337 std::vector<numpy::CharArray<32>> arrays;
1338 arrays.push_back(numpy::char_::array<32>({ "a", "b", "c" }));
1339 arrays.push_back(numpy::char_::array<32>({ "x", "y", "z" }));
1340
1341 auto result = numpy::char_::join("-", arrays);
1342
1343 bool passed = true;
1344
1345 if (!passed) {
1346 std::cout << " [FAIL] : in np_test_string_join() : join operation failed";
1347 throw std::runtime_error("np_test_string_join failed");
1348 }
1349
1350 std::cout << " -> tests passed" << std::endl;
1351 }
left (np_test_1_all.cpp:21799)
21789 }
21790
21791 void testRecFunctionsJoin() {
21792 std::cout << "========= testRecFunctionsJoin =======================";
21793
21794 auto dtype = std::make_shared<StructuredDType>(std::vector<std::pair<std::string, DType>>{
21795 {"key", DType::INT32},
21796 { "data", DType::FLOAT64 }
21797 });
21798
21799 RecordArray left(dtype, { 3 });
21800 left.setFieldValue({ 0 }, "key", static_cast<int32>(1));
21801 left.setFieldValue({ 0 }, "data", static_cast<float64>(10.0));
21802 left.setFieldValue({ 1 }, "key", static_cast<int32>(2));
21803 left.setFieldValue({ 1 }, "data", static_cast<float64>(20.0));
21804 left.setFieldValue({ 2 }, "key", static_cast<int32>(3));
21805 left.setFieldValue({ 2 }, "data", static_cast<float64>(30.0));
21806
21807 auto dtype2 = std::make_shared<StructuredDType>(std::vector<std::pair<std::string, DType>>{
21808 {"key", DType::INT32},
21809 { "info", DType::UNICODE32 }
length (np_test_1_all.cpp:7335)
7325 std::cout << " -> tests passed" << std::endl;
7326}
7327
7328void testStringTruncationStringTypes() {
7329 std::cout << "========= testStringTruncationStringTypes =======================";
7330
7331 // Test truncation behavior
7332 str8 small("This string is way too long for 8 characters");
7333 // std::cout << "Long string in str8: '" << small << "'" << std::endl;
7334 // std::cout << "Length: " << small.length() << " (max: " << str8::max_size << ")" << std::endl;
7335
7336 str16 medium("This is a medium length string that should be truncated");
7337 // std::cout << "Medium string in str16: '" << medium << "'" << std::endl;
7338 // std::cout << "Length: " << medium.length() << " (max: " << str16::max_size << ")" << std::endl;
7339
7340 std::cout << " -> tests passed" << std::endl;
7341}
7342
7343void testStringComparisonsStringTypes() {
7344 std::cout << "========= testStringComparisonsStringTypes =======================";
result (np_test_1_all.cpp:6090)
6080 throw std::runtime_error(description);
6081 }
6082 // std::cout << "Timer restart test passed. Second: " << second_elapsed << " ms" << std::endl;
6083
6084 std::cout << " -> tests passed" << std::endl;
6085}
6086
6087void test_benchmark_result_storageBenchmarkSorting() {
6088 std::cout << "========= test_benchmark_result_storage =======================";
6089
6090 BenchmarkResult result("TestAlgorithm", 1000, 15.5, true, 4096);
6091
6092 if (!(result.algorithm_name == "TestAlgorithm")) {
6093 std::string description = std::string("test_benchmark_result_storageBenchmarkSorting():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(result.algorithm_name == \"TestAlgorithm\")";
6094 std::cout << std::string("[FAIL] ") + description << std::endl;
6095 throw std::runtime_error(description);
6096 }
6097 if (!(result.array_size == 1000)) {
6098 std::string description = std::string("test_benchmark_result_storageBenchmarkSorting():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(result.array_size == 1000)";
6099 std::cout << std::string("[FAIL] ") + description << std::endl;
6100 throw std::runtime_error(description);
right (np_test_1_all.cpp:21812)
21802 left.setFieldValue({ 1 }, "key", static_cast<int32>(2));
21803 left.setFieldValue({ 1 }, "data", static_cast<float64>(20.0));
21804 left.setFieldValue({ 2 }, "key", static_cast<int32>(3));
21805 left.setFieldValue({ 2 }, "data", static_cast<float64>(30.0));
21806
21807 auto dtype2 = std::make_shared<StructuredDType>(std::vector<std::pair<std::string, DType>>{
21808 {"key", DType::INT32},
21809 { "info", DType::UNICODE32 }
21810 });
21811
21812 RecordArray right(dtype2, { 3 });
21813 right.setFieldValue({ 0 }, "key", static_cast<int32>(1));
21814 right.setFieldValue({ 0 }, "info", std::string("A"));
21815 right.setFieldValue({ 1 }, "key", static_cast<int32>(2));
21816 right.setFieldValue({ 1 }, "info", std::string("B"));
21817 right.setFieldValue({ 2 }, "key", static_cast<int32>(4));
21818 right.setFieldValue({ 2 }, "info", std::string("D"));
21819
21820 // std::cout << "Left array:" << std::endl;
21821 //left.printArray();
shift (np_test_1_all.cpp:2761)
2751 throw std::runtime_error(description);
2752 }
2753 if (!(left_array.getElementAt({3}) == 16384)) {
2754 std::string description = std::string("testBitShifts():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(left_array.getElementAt({3}) == 16384)";
2755 std::cout << std::string("[FAIL] ") + description << std::endl;
2756 throw std::runtime_error(description);
2757 }
2758
2759 // std::cout << "[OK] Left shift by array\n";
2760
2761 // Test excessive shift (should result in 0)
2762 auto excessive_left = left_shift(arr, 20); // More than 16 bits
2763 if (!(excessive_left.getElementAt({0}) == 0)) {
2764 std::string description = std::string("testBitShifts():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(excessive_left.getElementAt({0}) == 0)";
2765 std::cout << std::string("[FAIL] ") + description << std::endl;
2766 throw std::runtime_error(description);
2767 }
2768 if (!(excessive_left.getElementAt({1}) == 0)) {
2769 std::string description = std::string("testBitShifts():") + __FILE__ + ":" + std::to_string(__LINE__) + ": !(excessive_left.getElementAt({1}) == 0)";
2770 std::cout << std::string("[FAIL] ") + description << std::endl;
2771 throw std::runtime_error(description);