| Total Complexity | 129 |
| Total Lines | 405 |
| Duplicated Lines | 0 % |
Complex classes like Orange.tests.sql.FilterStringTest often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
| 1 | import unittest |
||
| 317 | @sql_test |
||
| 318 | class FilterStringTest(PostgresTest): |
||
| 319 | def setUp(self): |
||
| 320 | self.data = [ |
||
| 321 | [w] for w in "Lorem ipsum dolor sit amet, consectetur adipiscing" |
||
| 322 | "elit. Vestibulum vel dolor nulla. Etiam elit lectus, mollis nec" |
||
| 323 | "mattis sed, pellentesque in turpis. Vivamus non nisi dolor. Etiam" |
||
| 324 | "lacinia dictum purus, in ullamcorper ante vulputate sed. Nullam" |
||
| 325 | "congue blandit elementum. Donec blandit laoreet posuere. Proin" |
||
| 326 | "quis augue eget tortor posuere mollis. Fusce vestibulum bibendum" |
||
| 327 | "neque at convallis. Donec iaculis risus volutpat malesuada" |
||
| 328 | "vehicula. Ut cursus tempor massa vulputate lacinia. Pellentesque" |
||
| 329 | "eu tortor sed diam placerat porttitor et volutpat risus. In" |
||
| 330 | "vulputate rutrum lacus ac sagittis. Suspendisse interdum luctus" |
||
| 331 | "sem auctor commodo.".split(' ')] + [[None], [None]] |
||
| 332 | self.conn, self.table_name = self.create_sql_table(self.data) |
||
| 333 | self.table = SqlTable(self.conn, self.table_name) |
||
| 334 | |||
| 335 | def tearDown(self): |
||
| 336 | self.drop_sql_table(self.table_name) |
||
| 337 | |||
| 338 | def test_filter_string_is_defined(self): |
||
| 339 | filtered_data = filter.Values(conditions=[ |
||
| 340 | filter.FilterString(-1, filter.FilterString.IsDefined) |
||
| 341 | ])(self.table) |
||
| 342 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 343 | for row in self.data if row[0] is not None] |
||
| 344 | |||
| 345 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 346 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 347 | |||
| 348 | def test_filter_string_equal(self): |
||
| 349 | filtered_data = filter.Values(conditions=[ |
||
| 350 | filter.FilterString(-1, filter.FilterString.Equal, 'in') |
||
| 351 | ])(self.table) |
||
| 352 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 353 | for row in self.data if row[0] == 'in'] |
||
| 354 | |||
| 355 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 356 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 357 | |||
| 358 | def test_filter_string_equal_case_insensitive_value(self): |
||
| 359 | filtered_data = filter.Values(conditions=[ |
||
| 360 | filter.FilterString(-1, filter.FilterString.Equal, 'In', |
||
| 361 | case_sensitive=False) |
||
| 362 | ])(self.table) |
||
| 363 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 364 | for row in self.data if row[0] == 'in'] |
||
| 365 | |||
| 366 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 367 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 368 | |||
| 369 | def test_filter_string_equal_case_insensitive_data(self): |
||
| 370 | filtered_data = filter.Values(conditions=[ |
||
| 371 | filter.FilterString(-1, filter.FilterString.Equal, 'donec', |
||
| 372 | case_sensitive=False) |
||
| 373 | ])(self.table) |
||
| 374 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 375 | for row in self.data if row[0] == 'Donec'] |
||
| 376 | |||
| 377 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 378 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 379 | |||
| 380 | def test_filter_string_not_equal(self): |
||
| 381 | filtered_data = filter.Values(conditions=[ |
||
| 382 | filter.FilterString(-1, filter.FilterString.NotEqual, 'in') |
||
| 383 | ])(self.table) |
||
| 384 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 385 | for row in self.data if row[0] != 'in'] |
||
| 386 | |||
| 387 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 388 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 389 | |||
| 390 | def test_filter_string_not_equal_case_insensitive_value(self): |
||
| 391 | filtered_data = filter.Values(conditions=[ |
||
| 392 | filter.FilterString(-1, filter.FilterString.NotEqual, 'In', |
||
| 393 | case_sensitive=False) |
||
| 394 | ])(self.table) |
||
| 395 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 396 | for row in self.data if row[0] != 'in'] |
||
| 397 | |||
| 398 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 399 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 400 | |||
| 401 | def test_filter_string_not_equal_case_insensitive_data(self): |
||
| 402 | filtered_data = filter.Values(conditions=[ |
||
| 403 | filter.FilterString(-1, filter.FilterString.NotEqual, 'donec', |
||
| 404 | case_sensitive=False) |
||
| 405 | ])(self.table) |
||
| 406 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 407 | for row in self.data if row[0] != 'Donec'] |
||
| 408 | |||
| 409 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 410 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 411 | |||
| 412 | def test_filter_string_less(self): |
||
| 413 | filtered_data = filter.Values(conditions=[ |
||
| 414 | filter.FilterString(-1, filter.FilterString.Less, 'A') |
||
| 415 | ])(self.table) |
||
| 416 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 417 | for row in self.data |
||
| 418 | if row[0] is not None and row[0] < 'A'] |
||
| 419 | |||
| 420 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 421 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 422 | |||
| 423 | def test_filter_string_less_case_insensitive_value(self): |
||
| 424 | filtered_data = filter.Values(conditions=[ |
||
| 425 | filter.FilterString(-1, filter.FilterString.Less, 'In', |
||
| 426 | case_sensitive=False) |
||
| 427 | ])(self.table) |
||
| 428 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 429 | for row in self.data |
||
| 430 | if row[0] is not None and row[0].lower() < 'in'] |
||
| 431 | |||
| 432 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 433 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 434 | |||
| 435 | def test_filter_string_less_case_insensitive_data(self): |
||
| 436 | filtered_data = filter.Values(conditions=[ |
||
| 437 | filter.FilterString(-1, filter.FilterString.Less, 'donec', |
||
| 438 | case_sensitive=False) |
||
| 439 | ])(self.table) |
||
| 440 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 441 | for row in self.data |
||
| 442 | if row[0] is not None and row[0].lower() < 'donec'] |
||
| 443 | |||
| 444 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 445 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 446 | |||
| 447 | def test_filter_string_less_equal(self): |
||
| 448 | filtered_data = filter.Values(conditions=[ |
||
| 449 | filter.FilterString(-1, filter.FilterString.LessEqual, 'A') |
||
| 450 | ])(self.table) |
||
| 451 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 452 | for row in self.data |
||
| 453 | if row[0] is not None and row[0] <= 'A'] |
||
| 454 | |||
| 455 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 456 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 457 | |||
| 458 | def test_filter_string_less_equal_case_insensitive_value(self): |
||
| 459 | filtered_data = filter.Values(conditions=[ |
||
| 460 | filter.FilterString(-1, filter.FilterString.LessEqual, 'In', |
||
| 461 | case_sensitive=False) |
||
| 462 | ])(self.table) |
||
| 463 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 464 | for row in self.data |
||
| 465 | if row[0] is not None and row[0].lower() <= 'in'] |
||
| 466 | |||
| 467 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 468 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 469 | |||
| 470 | def test_filter_string_less_equal_case_insensitive_data(self): |
||
| 471 | filtered_data = filter.Values(conditions=[ |
||
| 472 | filter.FilterString(-1, filter.FilterString.LessEqual, 'donec', |
||
| 473 | case_sensitive=False) |
||
| 474 | ])(self.table) |
||
| 475 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 476 | for row in self.data |
||
| 477 | if row[0] is not None and row[0].lower() <= 'donec'] |
||
| 478 | |||
| 479 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 480 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 481 | |||
| 482 | def test_filter_string_greater(self): |
||
| 483 | filtered_data = filter.Values(conditions=[ |
||
| 484 | filter.FilterString(-1, filter.FilterString.Greater, 'volutpat') |
||
| 485 | ])(self.table) |
||
| 486 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 487 | for row in self.data |
||
| 488 | if row[0] is not None and row[0] > 'volutpat'] |
||
| 489 | |||
| 490 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 491 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 492 | |||
| 493 | def test_filter_string_greater_case_insensitive_value(self): |
||
| 494 | filtered_data = filter.Values(conditions=[ |
||
| 495 | filter.FilterString(-1, filter.FilterString.Greater, 'In', |
||
| 496 | case_sensitive=False) |
||
| 497 | ])(self.table) |
||
| 498 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 499 | for row in self.data |
||
| 500 | if row[0] is not None and row[0].lower() > 'in'] |
||
| 501 | |||
| 502 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 503 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 504 | |||
| 505 | def test_filter_string_greater_case_insensitive_data(self): |
||
| 506 | filtered_data = filter.Values(conditions=[ |
||
| 507 | filter.FilterString(-1, filter.FilterString.Greater, 'donec', |
||
| 508 | case_sensitive=False) |
||
| 509 | ])(self.table) |
||
| 510 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 511 | for row in self.data |
||
| 512 | if row[0] is not None and row[0].lower() > 'donec'] |
||
| 513 | |||
| 514 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 515 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 516 | |||
| 517 | def test_filter_string_greater_equal(self): |
||
| 518 | filtered_data = filter.Values(conditions=[ |
||
| 519 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'volutpat') |
||
| 520 | ])(self.table) |
||
| 521 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 522 | for row in self.data |
||
| 523 | if row[0] is not None and row[0] >= 'volutpat'] |
||
| 524 | |||
| 525 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 526 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 527 | |||
| 528 | def test_filter_string_greater_equal_case_insensitive_value(self): |
||
| 529 | filtered_data = filter.Values(conditions=[ |
||
| 530 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'In', |
||
| 531 | case_sensitive=False) |
||
| 532 | ])(self.table) |
||
| 533 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 534 | for row in self.data |
||
| 535 | if row[0] is not None and row[0].lower() >= 'in'] |
||
| 536 | |||
| 537 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 538 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 539 | |||
| 540 | def test_filter_string_greater_equal_case_insensitive_data(self): |
||
| 541 | filtered_data = filter.Values(conditions=[ |
||
| 542 | filter.FilterString(-1, filter.FilterString.GreaterEqual, 'donec', |
||
| 543 | case_sensitive=False) |
||
| 544 | ])(self.table) |
||
| 545 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 546 | for row in self.data |
||
| 547 | if row[0] is not None and row[0].lower() >= 'donec'] |
||
| 548 | |||
| 549 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 550 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 551 | |||
| 552 | def test_filter_string_between(self): |
||
| 553 | filtered_data = filter.Values(conditions=[ |
||
| 554 | filter.FilterString(-1, filter.FilterString.Between, 'a', 'c') |
||
| 555 | ])(self.table) |
||
| 556 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 557 | for row in self.data |
||
| 558 | if row[0] is not None and 'a' <= row[0] <= 'c'] |
||
| 559 | |||
| 560 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 561 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 562 | |||
| 563 | def test_filter_string_between_case_insensitive_value(self): |
||
| 564 | filtered_data = filter.Values(conditions=[ |
||
| 565 | filter.FilterString(-1, filter.FilterString.Between, 'I', 'O', |
||
| 566 | case_sensitive=False) |
||
| 567 | ])(self.table) |
||
| 568 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 569 | for row in self.data |
||
| 570 | if row[0] is not None and 'i' < row[0].lower() <= 'o'] |
||
| 571 | |||
| 572 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 573 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 574 | |||
| 575 | def test_filter_string_between_case_insensitive_data(self): |
||
| 576 | filtered_data = filter.Values(conditions=[ |
||
| 577 | filter.FilterString(-1, filter.FilterString.Between, 'i', 'O', |
||
| 578 | case_sensitive=False) |
||
| 579 | ])(self.table) |
||
| 580 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 581 | for row in self.data |
||
| 582 | if row[0] is not None and 'i' <= row[0].lower() <= 'o'] |
||
| 583 | |||
| 584 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 585 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 586 | |||
| 587 | def test_filter_string_contains(self): |
||
| 588 | filtered_data = filter.Values(conditions=[ |
||
| 589 | filter.FilterString(-1, filter.FilterString.Contains, 'et') |
||
| 590 | ])(self.table) |
||
| 591 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 592 | for row in self.data |
||
| 593 | if row[0] is not None and 'et' in row[0]] |
||
| 594 | |||
| 595 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 596 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 597 | |||
| 598 | def test_filter_string_contains_case_insensitive_value(self): |
||
| 599 | filtered_data = filter.Values(conditions=[ |
||
| 600 | filter.FilterString(-1, filter.FilterString.Contains, 'eT', |
||
| 601 | case_sensitive=False) |
||
| 602 | ])(self.table) |
||
| 603 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 604 | for row in self.data |
||
| 605 | if row[0] is not None and 'et' in row[0].lower()] |
||
| 606 | |||
| 607 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 608 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 609 | |||
| 610 | def test_filter_string_contains_case_insensitive_data(self): |
||
| 611 | filtered_data = filter.Values(conditions=[ |
||
| 612 | filter.FilterString(-1, filter.FilterString.Contains, 'do', |
||
| 613 | case_sensitive=False) |
||
| 614 | ])(self.table) |
||
| 615 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 616 | for row in self.data |
||
| 617 | if row[0] is not None and 'do' in row[0].lower()] |
||
| 618 | |||
| 619 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 620 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 621 | |||
| 622 | def test_filter_string_outside(self): |
||
| 623 | filtered_data = filter.Values(conditions=[ |
||
| 624 | filter.FilterString(-1, filter.FilterString.Outside, 'am', 'di') |
||
| 625 | ])(self.table) |
||
| 626 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 627 | for row in self.data |
||
| 628 | if row[0] is not None and not 'am' < row[0] < 'di'] |
||
| 629 | |||
| 630 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 631 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 632 | |||
| 633 | def test_filter_string_outside_case_insensitive(self): |
||
| 634 | filtered_data = filter.Values(conditions=[ |
||
| 635 | filter.FilterString(-1, filter.FilterString.Outside, 'd', 'k', |
||
| 636 | case_sensitive=False) |
||
| 637 | ])(self.table) |
||
| 638 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 639 | for row in self.data |
||
| 640 | if row[0] is not None and not 'd' < row[0].lower() < 'k'] |
||
| 641 | |||
| 642 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 643 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 644 | |||
| 645 | def test_filter_string_starts_with(self): |
||
| 646 | filtered_data = filter.Values(conditions=[ |
||
| 647 | filter.FilterString(-1, filter.FilterString.StartsWith, 'D') |
||
| 648 | ])(self.table) |
||
| 649 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 650 | for row in self.data |
||
| 651 | if row[0] is not None and row[0].startswith('D')] |
||
| 652 | |||
| 653 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 654 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 655 | |||
| 656 | def test_filter_string_starts_with_case_insensitive(self): |
||
| 657 | filtered_data = filter.Values(conditions=[ |
||
| 658 | filter.FilterString(-1, filter.FilterString.StartsWith, 'D', |
||
| 659 | case_sensitive=False) |
||
| 660 | ])(self.table) |
||
| 661 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 662 | for row in self.data |
||
| 663 | if row[0] is not None |
||
| 664 | and row[0].lower().startswith('d')] |
||
| 665 | |||
| 666 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 667 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 668 | |||
| 669 | def test_filter_string_ends_with(self): |
||
| 670 | filtered_data = filter.Values(conditions=[ |
||
| 671 | filter.FilterString(-1, filter.FilterString.EndsWith, 's') |
||
| 672 | ])(self.table) |
||
| 673 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 674 | for row in self.data |
||
| 675 | if row[0] is not None and row[0].endswith('s')] |
||
| 676 | |||
| 677 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 678 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 679 | |||
| 680 | def test_filter_string_ends_with_case_insensitive(self): |
||
| 681 | filtered_data = filter.Values(conditions=[ |
||
| 682 | filter.FilterString(-1, filter.FilterString.EndsWith, 'S', |
||
| 683 | case_sensitive=False) |
||
| 684 | ])(self.table) |
||
| 685 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 686 | for row in self.data |
||
| 687 | if row[0] is not None |
||
| 688 | and row[0].lower().endswith('s')] |
||
| 689 | |||
| 690 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 691 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 692 | |||
| 693 | def test_filter_string_list(self): |
||
| 694 | filtered_data = filter.Values(conditions=[ |
||
| 695 | filter.FilterStringList(-1, ['et', 'in']) |
||
| 696 | ])(self.table) |
||
| 697 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 698 | for row in self.data if row[0] in ['et', 'in']] |
||
| 699 | |||
| 700 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 701 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 702 | |||
| 703 | def test_filter_string_list_case_insensitive_value(self): |
||
| 704 | filtered_data = filter.Values(conditions=[ |
||
| 705 | filter.FilterStringList(-1, ['Et', 'In'], case_sensitive=False) |
||
| 706 | ])(self.table) |
||
| 707 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 708 | for row in self.data if row[0] in ['et', 'in']] |
||
| 709 | |||
| 710 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 711 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 712 | |||
| 713 | def test_filter_string_list_case_insensitive_data(self): |
||
| 714 | filtered_data = filter.Values(conditions=[ |
||
| 715 | filter.FilterStringList(-1, ['donec'], case_sensitive=False) |
||
| 716 | ])(self.table) |
||
| 717 | correct_data = [SqlRowInstance(filtered_data.domain, row) |
||
| 718 | for row in self.data if row[0] in ['Donec']] |
||
| 719 | |||
| 720 | self.assertEqual(len(filtered_data), len(correct_data)) |
||
| 721 | self.assertSequenceEqual(filtered_data, correct_data) |
||
| 722 |